This API has the following methods:
get_pruning_runner(model, inputs, method)
- A torch.nn.Module object to prune.
- A single or a list of torch.Tensor used as inputs for model inference. It need not be real data. It can be a randomly generated tensor of the same shape and data type as the real data.
- Either be 'iterative' or 'one_step'.