鉴于pytorch在科学论文领域用的越来越多,想写一个系列文章,讲述pytorch的用法。
要学习pytorch,一个前提是 知道python calss中的__call__
和__init__
方法.
简单的说就是:
__init__
: 类的初始化函数,类似于c++的构造函数__call___
: 使得类对象具有类似函数的功能。
__init__
比较好理解,现在主要看一下 __call__
的功能示例:
class A():
def __call__(self):
print('i can be called like a function')
a = A()
a()
out:
i can be called like a function
让我们在调用时传入参数如何?
class A():
def __call__(self, param):
print