GPU for Examples_core.ipynb

[betterze]

Dear PAIR code group,

Thank you for sharing this great work. I really like it.

I am using the Examples_core.ipynb now. Is there a way to use GPU in this torch implementation? The cpu one is a bit slow.

Thank you for your help.

Best Wishes,

Zongze