创建tensor
创建tensor有很多种方法,常见的如在numpy中创建后转为tensor,或者用形如torch.randn(2, 2)的函数创建tensor。常见创建tensor的方法如下:
# import from numpy
a = np.array([2, 3.3])
print(torch.from_numpy(a))
a = np.ones([2, 3])
print(torch.from_numpy(a))
# import from List
a = torch.tensor([2., 3.2])
print(a)
a = torch.FloatTensor(2, 2, 2)
print(a)
# uninitialized data
a = torch.empty([2, 2])
print(a)
a = torch.Tensor(2, 3)
print(a)
# get tensor's type
a = torch.tensor([1., 3])
print(a)
print(a.type())
print(a.size())
print(len(a.size()))
print(a.dim())
# create rand tensor
a = torch.rand(2, 2)
print(a)
b = torch.rand_like(a)
print(b)
a = torch.randint(1, 10, [2, 2])
print(a)
# create randn tensor
a = torch.randn(3, 3)
print(a)
# create full tensor
a = torch.full([2, 3], 7)
print(a)
print(torch.full([], 7.))
print(torch.full([1], 7.))
# arange
a = torch.arange(0, 10, 2)
print(a)
# matrix
print(torch.ones(2, 2))
print(torch.zeros(2, 2))
print(torch.eye(2, 2))
a = torch.eye(2)
print(torch.ones_like(a))