摘要: nn.Module() https://zhuanlan.zhihu.com/p/340453841 nn.Module() nn.Module是nn中十分重要的类,包含网络各层的定义及forward方法。 pytorch 里面一切自定义操作基本上都是继承nn.Module类来实现的。 简单的说 t 阅读全文
posted @ 2021-09-27 09:39 贝壳里的星海 阅读(4000) 评论(0) 推荐(1) 编辑