Skip to content

Is it possible to perform transforms.Spacing on gpu? #1521

Answered by ericspod
grudloff asked this question in Q&A
Discussion options

You must be logged in to vote

Spacing will work using CUDA if the input tensor is on a GPU device already. It should give a speed up if the rest of your transform sequence is optimal, the usual places where time is list is converting tensors or moving them to and from devices so avoiding these is important.

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by grudloff
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants