Cuda version of torch.eye()

2

Hi,

The eye function takes a device= argument. You can specify "cuda" there. Or provide a device object.

2 Likes