不是concat的意思

import torcha = torch.ones([1,2])b = torch.ones([1,2])torch.stack([a,b],1)(0 ,.,.) =  1 1 1 1[torch.FloatTensor of size 1x2x2]

以上這篇對PyTorch torch.stack的實例講解就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持武林站長站。