Pytorch-view的用法 torch.view(参数a,参数b,...) 有的时候会出现torch.view(-1)或者torch.view(参数a,-1)这种情况。
在pytorch中view函数的作用为重构张量的维度,相当于numpy中resize()的功能,但是用法可能不太一样。如下例所示比如
import torch
a=torch.Tensor([[[1,2,3],[4,5,6]]])
b=torch.Tensor([1,2,3,4,5,6])
print(a.view(1,6))
print(b.view(1,6))
te
第三方账号登入
QQ 微博 微信