批处理轴是否包含在softmax图层的可用轴中？

Question

在softmax图层中指定轴时：

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

是否包括批次/示例尺寸？换句话说，如果我有一个内幕（例如dim1，dim2，要素）并且我想在dim1上使用softmax，那是axis=0还是axis=1？

Answer 1

是的，包括批次尺寸。对于您的情况，您需要axis=1。