学习pytorch中的一些笔记(1)

学习pytorch中的一些笔记(1)

一、torch.Size

torch.Size() 查看当前Tensor的大小，即维度。

在官方教程还有一句话，我觉得可能有用：

torch.Size是个元组，元组的操作都可以用

len(torch.Size([5,3]))

输出：
2

torch.Size([5,3])[0]
torch.Size([5,3])[1]
输出：
5
3

torch.Size([5,3])+torch.Size([5,4])
输出：
torch.Size([5, 3, 5, 4])

二、add与add_的区别(下划线的作用）

实现加法的方法有：
1：x+y,(x和y要为同纬度)
2：torch.add(x,y) 也可以定义一个新的张量，将结果存进去torch.add(x,y，out=result)
3.y.add(x)或者y.add_(x)
y.add(x)与y.add_(x)的区别在于：
在加了_之后，y.add_(x)执行完之后y的值会变化为结果，不加_，y不会变.
看官方文档描述：

如文档所说，其他运算加_也是如此。
代码测试如下：

x = torch.rand(5,3)
y = x

print(y.add(x))
print(y)
print(y.add_(x))
print(y)

输出：
tensor([[1.9755, 1.0825, 1.8444],
        [0.8396, 0.60