megengine.utils.network.NodeFilterName

class NodeFilterName(node_iter, pattern, ignorecase)[源代码]

参考 NodeFilter.name

方法

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 ,其中包含给定变量依赖的所有节点。

make_re(pattern[, ignorecase])

name(pattern[, ignorecase])

通过节点名称过滤。

not_type(node_type)

移除过滤器中所有指定类型的计算节点。

param_provider()

get ParamProvider oprs; shorthand for .type(ParamProvider)

type(node_type)

通过节点类型过滤。