megengine.utils.network.NodeFilterType

class NodeFilterType(node_iter, node_type)

参考 NodeFilter.type.

方法

as_count()	返回该迭代器的长度。
as_dict()	遍历迭代器并返回一个有序字典，其键为节点名称，值为节点对象。
as_list()	将迭代器中的内容以列表的形式表示并返回。
as_unique()	如果迭代器中只包含一个元素，返回该元素，否则报错。
check_type(node_type)	检查迭代器中的节点对象都为给定类型，不是则报错。
data_provider()	返回迭代器中所有的 DataProvider 节点对象，该方法是 .type(DataProvider) 的缩写。
has_input(var)	寻找迭代器中以给定变量为输入的节点并返回。
make_all_deps(*dest_vars)	创建一个 NodeFilter 对象，该对象包含所有给定 vars 依赖的算子。
name(pattern[, ignorecase])	通过节点名称筛选。
not_type(node_type)	过滤出不是给定类型的算子。
param_provider()	get ParamProvider oprs; shorthand for .type(ParamProvider)
type(node_type)	通过给定的算子类型来过滤。

上一页

megengine.utils.network.NodeFilter.type

下一页

megengine.utils.network.NodeFilterType.as_count