np.expand_dims

`np.expand_dims` 是 NumPy 中的一个函数，用于在数组的指定位置插入新的维度。具体而言，它会将指定位置之前的维度保持不变，而在指定位置插入一个新的维度。例如，对于一个形状为 (3, 4) 的二维数组，如果我们想要在其最后插入一个新的维度，那么可以使用 `np.expand_dims(array, axis=2)`，这样就会得到一个形状为 (3, 4, 1) 的三维数组。其中，`array` 是需要插入新维度的数组，`axis` 是插入新维度的位置，取值范围为 `[-(array.ndim+1), array.ndim]`，默认为插入一个新的最后一个维度。