ソフトマックス アクティベーションの Kerasドキュメントには、アクティベーションが適用される軸を指定できると記載されています。私のモデルは、n x k行列Mを出力することになっています。ここで、Mijはi番目の文字が記号jである確率です。
n = 7 # number of symbols in the ouput string (fixed)
k = len("0123456789") # the number of possible symbols
model = models.Sequential()
model.add(layers.Dense(16, activation='relu', input_shape=((N,))))
...
model.add(layers.Dense(n * k, activation=None))
model.add(layers.Reshape((n, k)))
model.add(layers.Dense(output_dim=n, activation='softmax(x, axis=1)'))
kソフトマックス アクティベーションの軸 (私の場合は軸) を正しく指定する方法がわからないため、コードの最後の行はコンパイルされません。