Pytorch 中怎样才能正确编写 forward( ) 函数

1 问题描述

在学习 Pytorch 中我们知道, 在有 forward( ) 方法的函数, 可以通过直接实例化后, 通过 实例化( ) 的方法直接调用 forward( ) 函数, 但是却遇到了报错 :

• 'test' object is not callable

• 'test' object has no attribute '_backward_hooks'

下面来解释原因

---

2 解决方法

2.1 'test' object is not callable

出现这个问题是因为在创建 class 时, 没有继承 torch.nn.Module

2.2 test' object has no attribute '_backward_hooks'

这个问题是, 在继承 torch.nn.Module 后, 没有在 __init__ 方法中, super(自己) 去继承父类的方