Keras 批次轴是否包含在softmax层的可用轴中？

在softmax层中指定轴时：

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

---

是否包含批次/示例维度？换句话说，如果在发动机罩下我有（例如，dim1、dim2、功能），并且我想在dim1上使用softmax，那么是

轴=0

还是

轴=1

？

是，批次尺寸包括在内。对于您的情况，您需要

axis=1