pytorch框架包含大量的Python魔术方法,理解这些方法有助于理解源码;
概念介绍:在Python中,所有的以双下划线 "__“包起来的方法,都统称为为"Magic Method”;例如类的初始化时候会用到的构造函数
__init__
就是一个魔术方法,他在类对象的创建时执行;
Python中魔术方法汇总
可以参考 Python中的魔术方法汇总
这里介绍一些常用的;
__len__
方法
在Python类中,定义__len__
方法后;使用len()
可以查看__len__
方法的返回结果;
例如:在下列这段代码中,在类中定义了__len__
方法后,可以对对象data_ob
使用len
方法,从而或得数据集的大小,将这个对象作为一个数据集列表来看待;调用len(obj)
相当于obj.__len__
from torch.utils.data