Carlos-Francisco Méndez-Cruz / curso-alianza-2024

Go to a project
Toggle navigation Toggle navigation pinning
cmendezc
6e7c210e 1 parent 1b726367

NPU examples

Inline Side-by-side
Showing 1 changed file with 8 additions and 0 deletions
import torch
import torch.npu
x = torch.randn(2, 2).npu()
y = torch.randn(2, 2).npu()
z = x.mm(y)
print(z)
\ No newline at end of file