• M1 Mac
• PyTorch 2.0.1
  torch.device('mps') かつ、input_sizeを2以上にすると以下のエラーが発生する。torch.device('cpu')、もしくはinput_sizeを1にすると以下のエラーは発生しない。

RuntimeError: Expected a proper Tensor but got None (or an undefined Tensor in C++) for argument #0 'grad_y'

解決策

PyTorchを2.1.0にアップグレード