python二维、三维、思维数组之间的转化

python二维、三维、思维数组之间的转化

import numpy as np

# 假设你有一个二维数组，形状为（32，10000）
gray = np.random.randint(0, 256, (32, 10000), dtype=np.uint8)

# 将二维数组变成三维数组，形状为（32，100，100）
image = np.reshape(gray, (32, 100, 100))

# 将三维数组变成四维数组，形状为（32，100，100，1）
image = np.expand_dims(image, axis=-1)

这里使用了 numpy 的 reshape 函数将二维数组转换为三维数组，并使用 expand_dims 函数将三维数组转换为四维数组，添加了一个维度表示通道数，现在可以将这个四维数组输入