Hi does this work for tensors on gpu as well? Or do I have to convert them to CPU before passing through compose class.
Hi does this work for tensors on gpu as well? Or do I have to convert them to CPU before passing through compose class.