python numpy.expand_dims的用法

版权声明：本文为博主原创文章，未经博主允许不得转载O(∩_∩)O https://blog.csdn.net/qq_38153833/article/details/88999046

直接分析就知道了

import numpy as np
a = np.array([[[1,2,3],[4,5,6]]])
print(a)
print(a.shape)

target_region = np.expand_dims(a, 0)
print(target_region)
print(target_region.shape)

target_region = np.expand_dims(a, 1)
print(target_region)
print(target_region.shape)


target_region = np.expand_dims(a, 2)
print(target_region)
print(target_region.shape)

target_region = np.expand_dims(a, 3)
print(target_region)
print(target_region.shape)

target_region = np.expand_dims(a, 6)
print(target_region)
print(target_region.shape)