torch.stack和torch.cat区别,结论torch.stack,类似并联,结果是在相应的维度会增加一维;所以要求每个tensor大小相等torch.cat类似串联, ...
確定! 回上一頁