nn

包含原生 torch 层特定符号信息和实现的模块。

类

SymDepth

表示深度的符号参数。

class SymDepth

基础：Symbol

表示深度的符号参数。

__init__(*args, max_depth, **kwargs)

构造函数。

disable(_memo=None)

禁用符号。

Parameters:

_memo (Set[Symbol] | None) –

Return type:

无

is_skippable(idx)

返回idx是否可跳过。

Parameters:

idx (int) –

Return type:

bool

link_to(sp_parent)

链接到另一个符号。

Parameters:

sp_parent (Symbol) –

Return type:

无

property max_depth: int

返回最大深度。

property min_depth: int

返回最小深度。

set_skippable(idx, val)

设置idx是否可跳过。

Parameters:

• idx (int) –

• val (bool) –

Return type:

无

property skippable_idxs: List[int]

返回可跳过的索引的排序列表。