numpy.squeeze()

删除shape是1的维度, 不改变数组值

import numpy

a = numpy.arange(10)
print(a)
>>> [0 1 2 3 4 5 6 7 8 9]
b = numpy.reshape(a, [1, 1, -1])
print(b)
>>> [[[0 1 2 3 4 5 6 7 8 9]]]
c = numpy.squeeze(a)
print(c)
>>> [0 1 2 3 4 5 6 7 8 9]