2021
02-27
02-27
pytorch __init__、forward与__call__的用法小结
1.介绍当我们使用pytorch来构建网络框架的时候,也会遇到和tensorflow(tensorflow__init__、build和call小结)类似的情况,即经常会遇到__init__、forward和call这三个互相搭配着使用,那么它们的主要区别又在哪里呢?1)__init__主要用来做参数初始化用,比如我们要初始化卷积的一些参数,就可以放到这里面,这点和tf里面的用法是一样的2)forward是表示一个前向传播,构建网络层的先后运算步骤3)__call__的功能其实和forward...
继续阅读 >