深度学习【27】pytorch Variable变量方法的使用

这边记录一些Variable变量相关的一些方法。

将一个numpy的张量转成Variable：

Vtensor = torch.autograd.Variable(torch.LongTensor(np.int64(np.asarray(nptensor))))##torch.LongTensor确定变量类型，根据需要更改

类似reshape操作：

out = output.view(batchsize,3,-1)#output=[batchsize,3,256,256]，变成[batchsize,3,256*256]的张量

类似numpy中按多个索引取值

#numpy
a = np.asarray([1,2,3,4,5])
indexList=np.asarray([3,2,0])
b = a[indexList] #b=[4,3,1]
#pytorch

b = a.index_select(2,indexList)#其中2表示要在那个坐标下提取，a是一个Variable变量，indexList是一个LongTensor类型的Variable变量

类似clip函数

b = a.clamp(0,1)#将变量a，clip 到0-1之间

detach()函数

确保误差不传播到使用detach的结点前期的图。

还有很多函数，如matmul，min，max等等。