__init__
功能说明
PrunneTorch类方法,对用户输入的模型进行类初始化。
函数原型
__init__(network, inputs)
参数说明
参数名 |
输入/返回值 |
含义 |
使用限制 |
---|---|---|---|
network |
输入 |
待剪枝模型实例。 |
必选。 数据类型:PyTorch模型。 |
inputs |
输入 |
模型的输入数据,用于解析模型。 |
可选。 数据类型:Tensor。 |
父主题: PruneTorch