Torch¶

import numpy as np
import torch

a = torch.Tensor([1, 2, 3])
b = torch.Tensor([4, 5, 6])
print(a + b)

c = torch.ones(2, 3)
d = torch.zeros(4, 5)
print(d.shape)
print(d.size())
e = d.numpy()
print(e)
print(torch.add(a, b))
print(torch.argmax(a))