torch.stack()

本文详细解析了PyTorch中stack()函数的使用方法,解释了如何在不同的维度上叠加Tensor,并通过实例说明了从外到内的观察视角在操作中的应用。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

torch.stack()

含义:在指定的维度对元素进行叠加

例子:

使用torch生成如下的三个tensor: a, b, c

在指定的维度 0 对a, b, c进行叠加:

在指定的维度 1 对a, b, c进行叠加:

在指定的维度 2 对a, b, c进行叠加:

注意: 对于维度 0, 1, 2应该从外往里面看进去. 在指定的维度对tensor进行叠加时, 应该把该维度对应的个体看成是一个块, 然后进行块的叠加. 例如, 0 维度对应的就是把每一个tensor看成是块, 进行叠加时,就把每一个tensor整体进行叠加. 当指定叠加的维度是 1 时,   就把每一个tensor从外往里看进去一层, 把该层对应的每一个个体看成一个块, 然后再三个 tensor 再对应叠加. 同理, 再往里看进去一层进行上述的操作即可得到维度为 2 的叠加.  

重点: 这里在不同维度对 tensor 进行叠加重点在于对tensor 从外往里看进去,然后把看到的那一层的每一个元素看成是块, 最后再把所有 tensor 的块进行叠加.

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值