在此示例中,我们将展平整个张量图像,但是如果我们只想展平张量内的特定轴怎么办?这是使用CNN时通常需要的操作。
让我们看看如何使用PyTorch展平代码中的张量的特定轴。...想知道stack() 方法是如何工作的吗?stack()方法的解释将在本系列的后面介绍。...2, 2, 2, 2, 2, 2,
2, 2, 2, 2, 2, 2, 2, 2, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3])
在底部,你会注意到作为张量对象的方法内置的另一种方式...检查形状,我们可以看到我们有一个2级张量,其中三个单色通道图像被展平为16个像素。
四、扁平化一个RGB图
如果我们将RGB图像展平,那么颜色会怎样?...每个颜色通道将首先被展平。然后,展平后的通道将在张量的单个轴上并排排列。让我们来看一个代码示例。
我们将构建一个示例RGB图像张量,高度为2,宽度为2。