2024年6月8日
pytorch小知识（01）：forward方法
摘要： 我们都知道torch.nn.Module在被继承时要求我们改写两个方法，一个是__init__，一个是forward。前者用于定义层，后者用于定义前向计算的流程。但是当我们在实际使用一个网络时，我们不会使用forward这个方法进行计算，而是进行如下的操作： 点击查看代码 import torch 阅读全文
posted @ 2024-06-08 07:32 小丑与锁鸟 阅读(229) 评论(0) 推荐(0)