模块

所有稀疏模块的动态类。

类

所有稀疏模块的基础动态类。

class SparseModule

基础：DynamicModule

所有稀疏模块的基础动态类。

modify(*args, **kwargs)

初始化稀疏掩码时调用此函数。

请注意，对于任何未通过规则中的None冻结的模块，将调用此函数。因此，我们仅在必要时使用此函数来初始化稀疏掩码。

set_mask(value)

设置模块权重的活动稀疏掩码。

Parameters:: value (BoolTensor | None) –