从今天起要从头开始学习PyTorch了,在此记下笔记。
PyTorch 入门第一步
Tensor
import Torch as t
x = t.tensor(5,3)#生成5*3的矩阵
print(x.size())# 输出x的维度
print(x.size()[0]) #输出x的第0维
print(x.size(0))# 输出x的第0维
y = t.rand(5,3)# 生成0-1之间的随机数矩阵
# 加法
x + y
t.add(x,y)
result = t.rand(5,3)
t.add(x,y,out=result)
inplace
y.add(x)#不改变y的值
y.add_(x)# 改变y的值
函数名带下划线的会修改Tensor本身,x.add(y),x.t()会返回一个新的Tensor,而x不变,但x.add_(y)改变x。
Tensor 与 numpy
tensor转numpy
a = t.tensor(5,3)
b = a.numpy()
numpy 转 tensor
import numpy as np
a = np.ones(5)
b = t.from_numpy(a)
转换之后,因为tensor和numpy共享内存,所以一个改变另一个也会被改变。
AutoGrad自动微分
Variable