step34.pyのflatten()について
kawase621 opened this issue · comments
kawase621 commented
step34.pyのコードで、y.dataとx.grad.dataにflatten()が実行されていますが、当該処理は必要でしょうか?
y及びx.gradが保持するdataはいずれも最初から1次元配列だと思うのですが。
Koki Saitoh commented
確かに、この flatten()
は必要ないですね。
こちら修正したいとおもいます。
ご指摘、ありがとうございます!