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