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