torch.stack与np.stack均是在已有的基础上创建一个新的维度,因此要求提供的矩阵/张量具有相同的维度。
torch.cat与np.concatenate则都是在已有的行列上进行拼接,必须保证需要拼接的方向具有相同的大小。
torch.cat/stack,np.concatenate/stack区分
最新推荐文章于 2023-10-18 10:28:15 发布
torch.stack与np.stack均是在已有的基础上创建一个新的维度,因此要求提供的矩阵/张量具有相同的维度。
torch.cat与np.concatenate则都是在已有的行列上进行拼接,必须保证需要拼接的方向具有相同的大小。