tensor_shape

tensor_shape 是一个列表，用于指定输入张量的形状。它有三个元素，分别表示输入张量的通道数、高度和宽度。例如，如果我们的输入张量形状为 [3, 224, 224]，表示它有三个通道，高度为224像素，宽度也为224像素。

在数据预处理管道中，我们通常需要将输入数据转换为指定形状的张量。如果输入数据是图像，我们可以先将其裁剪或缩放到指定的高度和宽度，然后将其转换为Numpy数组，并将通道维度置于第一个维度，即 [3, H, W]。最后，我们可以使用PyTorch中的torch.from_numpy()函数将Numpy数组转换为张量，并使用torch.unsqueeze()函数增加一个批次维度，即 [1, 3, H, W]。如果我们指定的tensor_shape为 [3, 224, 224]，则需要在转换为张量之前先将输入数据裁剪或缩放为 224x224 的大小。