如何使用numpy random创建随机数据的4D数组并适配(1,32,32,3)模型输入
生成指定形状的随机4D numpy数组
你可以直接利用numpy的随机数生成函数,传入目标形状参数即可得到符合要求的4D数组,以下是几种常用实现方式:
- 生成**[0,1)区间均匀分布**的随机数组:
import numpy as np # 生成形状为(1, 32, 32, 3)的随机数组 random_arr = np.random.rand(1, 32, 32, 3)
- 生成**标准正态分布(均值0,方差1)**的随机数组:
random_arr_normal = np.random.randn(1, 32, 32, 3)
- 生成指定范围的整数随机数组(比如模拟图像像素的0-255取值):
# 生成0到255之间的整数数组(左闭右开区间,因此上限设为256) random_arr_int = np.random.randint(0, 256, size=(1, 32, 32, 3))
验证数组形状是否符合要求:
print(random_arr.shape) # 输出 (1, 32, 32, 3)
内容的提问来源于stack exchange,提问作者Mahesh
相关产品推荐
相关产品推荐

