PyTorch中torch.stack的示例分析
这篇文章主要介绍了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的示例分析”这篇文章对大家有帮助，同时也希望大家多多支持亿速云，关注行业资讯频道，更多相关知识等着你来学习!