How to do elementwise multiplication of two vectors?

8

If both tensors are stored on the GPU, then yes.
The operations are written for CPU and GPU tensors, so as long as the data is pushed to the device, the GPU will be used.

7 Likes