Pytorch中返回super().forward()的用法

最新推荐文章于 2023-09-28 07:20:55 发布

怎样才能回到过去

最新推荐文章于 2023-09-28 07:20:55 发布

阅读量409

点赞数 1

分类专栏： Pytorch 中的各种函数文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/z2572862506/article/details/129967375

版权

Pytorch 中的各种函数专栏收录该内容

54 篇文章 1 订阅

订阅专栏

作用:返回父类中的forward()方法

import torch
 
 
class Parent(torch.nn.Module):
    def forward(self, tensor):
        return tensor + 1
 
 
class Child(Parent):
    def forward(self, tensor):
        return super(Child, self).forward(tensor) + 1
 
 
module = Child()
# Increment output by 1 so we should get `4`
module.register_forward_hook(lambda module, input, output: output + 1)
print(module(torch.tensor(1))) # and it is 4 indeed
print(module.forward(torch.tensor(1))) # here it is 3 still