Variable (torch.randn(3, 4)) loss = loss_fn(input, target) print(input) print(target) print(loss) print(input.size(), target.size(), ...
確定! 回上一頁