この API には次のメソッドがあります。
get_pruning_runner(model, inputs, method)- model
- プルーニングする
torch.nn.Moduleオブジェクト。 - inputs
- 1 つの torch、またはそのリスト。モデル推論の入力として使用されるテンソル。実際のデータは必要ありません。形状とデータ型が実際のデータと同じであれば、無作為に生成されるテンソルでもかまいません。
- method
- 'iterative' または 'one_step' のいずれか。