评分

提单

论坛

小AI

init

功能说明

PrunneTorch类方法，对用户输入的模型进行类初始化。

函数原型

__init__(network, inputs)

参数说明

参数名

输入/返回值

含义

使用限制

network

输入

待剪枝模型实例。

必选。

数据类型：PyTorch模型。

inputs

输入

模型的输入数据，用于解析模型。

可选。

数据类型：Tensor。

父主题： PruneTorch