Как наносить нулями тензор вдоль некоторой оси (Python)

Вы можете использовать np.pad() :

 a = np.ones((4, 3, 2)) # npad is a tuple of (n_before, n_after) for each dimension npad = ((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.]]]