quick_verification.py 108 Bytes Raw Blame History Permalink 1 2 3 4 5 6 7 8 import torch import torch.npu x = torch.randn(2, 2).npu() y = torch.randn(2, 2).npu() z = x.mm(y) print(z)
