这里写目录标题
tensor
创建
:
获取形状
:shape
或者size函数
加法
:torch.add(x,y,out=y)
,当然还有其他方式,但是这个方式最好!
索引
:与numpy类似
其余选择函数
:
改变形状
:view函数
,1、并且可以使用-1
表示系统自动推算,x=torch.randn(15).view(-1,5)
;2、view返回的tensor与原来的tensor共享内存,可以先clone在使用view,y=x.clonew().view(15)。
将tensor转换为python的number
:item函数
线代函数
:
tensor转numpy
:tensor.numpy()
numpy转tensor
:torch.from_numpy()
to函数
将Tensor你放到设备上