原文:http://pytorch.org/tutorials/beginner/blitz/tensor_tutorial.html
- 声明一个张量
- x = torch.Tensor(5,3)
生成一个5*3的为初始化的张量
- x = torch.rand(5,3)
生成一个5*3的全部元素随机的张量
- print(x.size())
打印x的大小,那肯定是5*3,但是他的显示形式如下,而且这也是一个元祖,那么也就是说是一个1*2的张量。原文:torch.Size
is in fact a uple, so it supports the same operations。
Torch.Size([5,3])
- 实现一个两个张量的加法,有四种实现方法:
- print(x + y)
- print(torch.add(x + y))
- result = torch.Tensor(5, 3)
torch.add(x, y, out = result)
print(result) - y.add_(x)
print(y)
那么可以看到1,2,3都没有改变x或者y的值,那么4通过下划线改变了y的值。
- 如果用numpy那么很重要的一点就是,numpy和pytorch的张量共用一个内存。用pytorch改了张量,numpy的array也会跟着改变。