Python numpy expand_dims пример функции
# welcome to softhunt.net
# import numpy
import numpy as np
# using Numpy.expand_dims() function
softhunt = np.array([[1, 2], [7, 8]])
print(softhunt.shape)
softhunt = np.expand_dims(softhunt, axis = 1)
print(softhunt.shape)
Outrageous Ostrich