NumPy中ndarray的shape参数4个值指什么？shape=(1,224,224,3)解析

1. NumPy中创建ndarray时，shape参数传入的4个数值分别代表什么？ #

• shape参数的每个数值对应ndarray一个维度上的元素数量，4个数值就表示这是一个四维数组。
• 维度的具体含义没有固定规则，完全取决于数据场景：比如在机器学习中常用来表示「样本数-特征维度-...」的层级，在计算机视觉中则对应「样本数-图像尺寸-通道数」这类结构，核心是每个位置的数字定义了对应维度的长度。

2. 解释shape=(1, 224, 224, 3)中每个数值的具体含义 #

这是计算机视觉任务里非常常见的张量形状，每个数值的含义如下：

• 1：代表样本数量，这里表示当前数组仅包含1张图像样本
• 224：代表图像的高度，即垂直方向上的像素数
• 224：代表图像的宽度，即水平方向上的像素数
• 3：代表图像的通道数，对应RGB彩色图像的3个颜色通道（红、绿、蓝）

内容的提问来源于stack exchange，提问作者sks