PyTorch 损失函数

test 1

test 2