megengine.utils.network.Network

class Network

属性

all_oprs
all_oprs_dict
all_vars
all_vars_dict
data_providers_filter	返回网络的输入。
dest_vars	返回网络的输出。
metadata	Load metadata as a dict.
opr_filter	创建并返回一个包含网络中所有计算节点的筛选器。
params_filter	筛选出网络中的所有参数，也就是 ImmutableTensor 类型的计算节点。
var_filter	返回一个筛选器，它会作用于网络中的所有变量。

方法

add_dep_oprs(*vars)
add_output(*vars)	把 vars 添加到网络的输出列表中。
dump(file, *[, keep_var_name, ...])	把网络保存为模型文件。
get_dep_oprs(var)	返回 var 所依赖的计算节点的集合。
get_opr_by_name(name[, unique])
get_opr_by_type(oprcls[, unique])
get_var_by_name(name[, unique])
get_var_receive_oprs(var)	返回所有以 var 为输入的计算节点。
load(model_path[, outspec])	加载模型并生成一个 Network 对象。
make_const(data[, name, device])	创建一个 ImmutableTensor 对象，为网络提供计算参数。
make_input_node(shape, dtype[, name, device])	创建一个 Host2DeviceCopy 对象，为网络提供输入。
modify_opr_names(modifier)	原地 修改计算节点的名称，通常用于在把一个网络合并入另一个网络时。
optimize_for_inference(dest_vars, **kwargs)	优化该网络，使其在推理时获得更优越的性能。
remove_output(*vars)	Removes vars from the network output node list
replace_oprs(repl_dict)	替换图中的算子。
replace_vars(repl_dict)	替换图中的变量。
reset_batch_size(batchsize, *[, blacklist])	重置批大小，该操作会作用于网络中不在黑名单内的所有变量。

上一页

megengine.utils.hook.HookHandler.remove

下一页

megengine.utils.network.Network.all_oprs