pytorch自定义新层demo

上传：fandejie 浏览： 35 推荐： 0 文件：PY 大小：1.78KB 上传时间：2019-01-10 13:37:09 版权申诉

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