torch.stack （）在这里插入图片描述一、准备数据二、dim=0三、dim=1四、dim=2 一、准备数据 首先把基本的数据准备好： import torch import numpy as np # 创建3*3的矩阵，a、b a=np.array([[1,2,3],[4,5,6],[7,8,9]]) b=np.array([[10,20,30],[40,50,60],[70,80,90]]) # 将矩阵转化为Tensor a = torch.from_numpy(a) b = tor