'''
#torch.stack((a,b),dim)沿着新的dim拼接a,b,大前提就是:a,b的形状要完全一样,拼接前a,b是三维的那么
#拼接后a,b就是四维的
from this import d
import torch
ysh=torch.rand((3,4,5))
mes=torch.zeros((3,4,5))
kaka=torch.stack((ysh,mes),dim=1)
print("kaka_shape",kaka.shape)
#kaka_shape dim
#[2, 3, 4, 5] 0
#[3, 2, 4, 5] 1
'''
torch.stack()
最新推荐文章于 2024-04-23 00:59:08 发布