以包含批量维度的图像张量为例,设张量 A 保存了 4 张,长和宽为 32 的三通道像素矩阵,则张量 A 的形状为 [4, 3, 32, 32](PyTorch将通道维度放在前面,即 (batch_size...同样的方式,张量 B 保存了另外 5 张,长和宽为 32 的三通道像素矩阵,张量 B 的形状为 [5, 3, 32, 32]。...现在需要在批量维度上合并两个包含批量维度的图像张量,这里批量维度索引号为 0,即 dim = 0,合并张量 A 和 B 的代码如下:
import torch
# 模拟图像张量A
a = torch.randn...比如图像张量形状为 [4, 3, 32, 32] 和形状为 [5, 1, 32, 32] 的张量不能直接在批量维度上进行合并,因为两个图像张量的通道维度的长度不一致,一个是单通道(channels =...《TensorFlow深度学习》