PyTorchでのcross-validation

PyTorchでcross-validation(交差検証。以下CV)する場合の実装例を参考として残しておきます。 まずは訓練時の実装例です。 次に推論時の実装例です。 cross-validationをすることで、しない場合に比べて計算量はk(=分割数)倍になりますが、モデルの評価がより確かなものになります。 参考まで。