np.expand_dims(A, axis=0)
np.argmax(E, axis=1)
np.newaxis
>> type(np.newaxis)
NoneType
>> np.newaxis == None
True
>> x = np.arange(3)
>> x
array([0, 1, 2])
>> x.shape
(3,)
>> x[:, np.newaxis]
array([[0],
[1],
[2]])
np.expand_dims(A, axis=0)
np.argmax(E, axis=1)
np.newaxis
>> type(np.newaxis)
NoneType
>> np.newaxis == None
True
>> x = np.arange(3)
>> x
array([0, 1, 2])
>> x.shape
(3,)
>> x[:, np.newaxis]
array([[0],
[1],
[2]])