softmax関数のテンソルフローで「dim」パラメータを使用するのはなぜですか？

Question

テンソルフローでsoftmax関数の 'dim'パラメータを使用するのはなぜですか？入力としてどのようなテンソルを使用できますか？

Answer 1

tf.nn.softmaxは、一般的な空でないテンソルを受け入れます。

softmaxを適用したいディメンションごとに決定することができます。

通常、softmaxは入力テンソルの最後の次元（デフォルト動作）に適用されます。これは通常softmaxがニューラルネットワークの出力に適用されるため、通常は[batch_size, num_classes]という形のテンソルです。

しかし、あなたは[batch_size, num_classes, 2, 1]の形状とテンソルにソフトマックスを適用することを決定したとだけテンソルの第2次元上のソフトマックスを計算することができます：tf.nn.softmax(tensor, axis=1)