のように0～9に対する確率10個セットがバッチサイズ個ある二次元配列になっているので(出力.size()をすると、(バッチサイズ, 10) と出てくるはずである)、引数でdim = 1と指定することで、バッチサイズ個ある確率10個セットそれぞれの中だけでソフトマックスを行います。こ...