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 = torch.from_numpy(b) # 打印a、b、c print(a) print(b) outp