Включена ли ось партии в доступные оси слоя softmax?

Question

При указании оси в слое softmax:

out = keras.layers.Softmax(axis=x)(in)

Включен ли размер партии / примера? Другими словами, если у меня под капотом (например, dim1, dim2, функции), и я хочу softmax над dim1, это axis=0 или axis=1?

Answer 1

Да, размер партии включен. Для вашей ситуации вы бы хотели axis=1.