get_dframe('q{}_{}') print('get_dframe') print(df) #1つ目の結果を自動配列で確認(image形式)(2次元のみ) img, subs = Auto_array(result[0]).get_image('q{}_{}') import matplotlib.pyplot as plt print('get_image') plt.imshow(img) plt.yticks(range(len(subs[0])), subs[0]) ...
のように0～9に対する確率10個セットがバッチサイズ個ある二次元配列になっているので(出力.size()をすると、(バッチサイズ, 10) と出てくるはずである)、引数でdim = 1と指定することで、バッチサイズ個ある確率10個セットそれぞれの中だけでソフトマックスを行います。こ...