PyTorch Forums
shape_mismatch