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的操作
- 有四种常用方式(以相加为例)
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)