pytorch学习（1） Tensor与Autograd

pytorch学习笔记（1） Tensor与Autograd

开个专栏记录一些Pytorch的常用用法，方便自己查找。
一下内容均来自pytorch基础教程

Tensor

tensor是每个深度学习框架都要用到的模块，在pytorch中，它有如下几种常见创建方法：

创建tensor

x = torch.empty(5, 3) # 未初始化的非零张量
x = torch.rand(5, 3)  # 随机初始化张量
x = torch.zeros(5, 3, dtype=torch.long) # 0值张量
x = torch.tensor([5.5, 3]) # 把数组转换为张量
x = torch.randn_like(x, dtype=torch.float) # 创建跟x同大小的张量

tensor的操作

1. 有四种常用方式（以相加为例）

x = torch.ones(2, 2)
y = torch.tensor([[1,2],[2,1]],dtype=torch.float)
result = torch.empty(2,2)
torch.add(x, y, out=result)
print(x+y)
print