本代码实现了pytorch自定义层的demo,包括层的定义和测试数据、模型训练、loss图像可视化