pytorch 学习--60分钟入个门

标量(Scalar)是只有大小,没有方向的量,如1,2,3等
向量(Vector)是有大小和方向的量,其实就是一串数字,如(1,2)
矩阵(Matrix)是好几个向量拍成一排合并而成的一堆数字,如[1,2;3,4]
其实标量,向量,矩阵它们三个也是张量,标量是零维的张量,向量是一维的张量,矩阵是二维的张量。

简单相加
a+b

torch.add(a,b)
给出一个输出向量
torch.add(a,b,out=result)
原地相加
b.add_(a) # 这时值已经给了 b

张量部分截取

print(y[:,1])

tensor 转 numpy

a = torch.rand(5,3)
a.numpy()

numpy 转 tensor

b = torch.from_numpy(a)

猜你喜欢

转载自www.cnblogs.com/Frank99/p/9474927.html