Pytorchのfor文内で定義した変数ってローカルスコープになるんだっけ？
VSCodeのpylanceでfor文の外で変数呼ぼうとすると未定義だって言われる
言語によってぐちゃぐちゃだからよくわからん