a的shape从(96,96)变成(1000,96,96)
可以使用:
np.expand_dims(a,0).repeat(1000,axis=0)
解释:
expand_dims表示增加一个维度,这个维度增加在a的0维度。
repeat代表重复的次数,axis代表在哪个维度进行重复。
可以根据自己的需要更改参数。
如果有更简便的方法欢迎留言!
a的shape从(96,96)变成(1000,96,96)
可以使用:
np.expand_dims(a,0).repeat(1000,axis=0)
expand_dims表示增加一个维度,这个维度增加在a的0维度。
repeat代表重复的次数,axis代表在哪个维度进行重复。
可以根据自己的需要更改参数。
如果有更简便的方法欢迎留言!