torch.stack

看网上看多没讲的不是很明白,我来试试空间上的理解

复制代码
# 假设是时间步T1的输出
T1 = torch.tensor([[1, 2, 3],
        		[4, 5, 6],
        		[7, 8, 9]])
# 假设是时间步T2的输出
T2 = torch.tensor([[10, 20, 30],
        		[40, 50, 60],
        		[70, 80, 90]])

输出:

复制代码
print(torch.stack((T1,T2),dim=0).shape)
print(torch.stack((T1,T2),dim=1).shape)
print(torch.stack((T1,T2),dim=2).shape)
print(torch.stack((T1,T2),dim=3).shape)
# outputs:
torch.Size([2, 3, 3])
torch.Size([3, 2, 3])
torch.Size([3, 3, 2])
'选择的dim>len(outputs),所以报错'
IndexError: Dimension out of range (expected to be in range of [-3, 2], but got 3)

将dim=0,1,2理解为x,y,z轴.

  1. 那么dim=0的拼接就是在x轴上:这么看,就是x=2,y=3,z=3
  2. dim = 1,这么看,x=3,y=2,z=3
  3. dim = 2,这么看,x=3,y=3,z=2

    我偷懒了,图就是意思意思
相关推荐
Juchecar8 分钟前
一文讲清 nn.Module 中 forward 函数被调用时机
人工智能
小彭律师21 分钟前
基于因果推理与Transformer的金融理财产品智能推荐系统
深度学习·金融·transformer
七牛云行业应用26 分钟前
深度解析强化学习(RL):原理、算法与金融应用
人工智能·算法·金融
说私域34 分钟前
“开源AI智能名片链动2+1模式S2B2C商城小程序”在直播公屏引流中的应用与效果
人工智能·小程序·开源
Hcoco_me1 小时前
深度学习和神经网络之间有什么区别?
人工智能·深度学习·神经网络
霍格沃兹_测试1 小时前
Ollama + Python 极简工作流
人工智能
资源开发与学习1 小时前
AI智时代:一节课带你玩转 Cursor,开启快速入门与实战之旅
人工智能
西安光锐软件1 小时前
深度学习之损失函数
人工智能·深度学习
补三补四1 小时前
LSTM 深度解析:从门控机制到实际应用
人工智能·rnn·lstm
astragin1 小时前
神经网络常见层速查表
人工智能·深度学习·神经网络