这篇文章主要介绍了PyTorch中torch.stack的示例分析，具有一定借鉴价值，感兴趣的朋友可以参考下，希望大家阅读完这篇文章之后大有收获，下面让小编带着大家一起了解一下。

不是concat的意思

import torch
a = 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的示例分析”这篇文章对大家有帮助，同时也希望大家多多支持天达云，关注天达云行业资讯频道，更多相关知识等着你来学习!