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
1 +import torch
2 +import torch.npu
3 +
4 +x = torch.randn(2, 2).npu()
5 +y = torch.randn(2, 2).npu()
6 +z = x.mm(y)
7 +
8 +print(z)
...\ No newline at end of file ...\ No newline at end of file