对PyTorch torch.stack的实例讲解
不是concat的意思
importtorch a=torch.ones([1,2]) b=torch.ones([1,2]) torch.stack([a,b],1) (0,.,.)= 11 11 [torch.FloatTensorofsize1x2x2]
以上这篇对PyTorchtorch.stack的实例讲解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持毛票票。
不是concat的意思
importtorch a=torch.ones([1,2]) b=torch.ones([1,2]) torch.stack([a,b],1) (0,.,.)= 11 11 [torch.FloatTensorofsize1x2x2]
以上这篇对PyTorchtorch.stack的实例讲解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持毛票票。