torch.cuda
package torch.cuda
This package adds support for CUDA tensor types, that implement the same function as CPU tensors, but they utilize GPUs for computation.
Attributes
Members list
In this article
This package adds support for CUDA tensor types, that implement the same function as CPU tensors, but they utilize GPUs for computation.