np.expand_dims(data, axis=int)增加数据维度

增加数据维度 new_data = np.expand_dims(data, axis=int)
参数axis表示:在哪里加一个维度,其对应的是data.shape中的位置索引(axis<=len(data.shape)),
0表示最前,-1表示最末。
假设原始数据 data.shape = (28,28,1),则:
axis = 0时:(28,28,1)🔜 (1,28,28,1)
axis = 1时:(28,28,1)🔜 (28,1,28,1)
axis = -1时:(28,28,1)🔜 (28,28,1,1)
axis = 4时:会报错,超出了原来的最大维度个数

也可以使用reshape方法达到相同的目的,例如:

data.reshape((data.shape[0], data.shape[1], data.shape[2], 1))
# 就等价与
np.expand_dims(data, axis=-1)
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值