您可以使用
np.pad()：
a = np.ones((4, 3, 2))# npad is a tuple of (n_before, n_after) for each dimensionnpad = ((0, 0), (1, 2), (2, 1))b = np.pad(a, pad_width=npad, mode='constant', constant_values=0)print(b.shape)# (4, 6, 5)print(b)# [[[ 0. 0. 0. 0. 0.]# [ 0. 0. 1. 1. 0.]# [ 0. 0. 1. 1. 0.]# [ 0. 0. 1. 1. 0.]# [ 0. 0. 0. 0. 0.]# [ 0. 0. 0. 0. 0.]]# [[ 0. 0. 0. 0. 0.]# [ 0. 0. 1. 1. 0.]# [ 0. 0. 1. 1. 0.]# [ 0. 0. 1. 1. 0.]# [ 0. 0. 0. 0. 0.]# [ 0. 0. 0. 0. 0.]]# [[ 0. 0. 0. 0. 0.]# [ 0. 0. 1. 1. 0.]# [ 0. 0. 1. 1. 0.]# [ 0. 0. 1. 1. 0.]# [ 0. 0. 0. 0. 0.]# [ 0. 0. 0. 0. 0.]]# [[ 0. 0. 0. 0. 0.]# [ 0. 0. 1. 1. 0.]# [ 0. 0. 1. 1. 0.]# [ 0. 0. 1. 1. 0.]# [ 0. 0. 0. 0. 0.]# [ 0. 0. 0. 0. 0.]]]
