dmx.compressor.functional.approximate.TorchFunctionApproximation

class dmx.compressor.functional.approximate.TorchFunctionApproximation(func_id: None, algorithm: str = 'vsimd', wrapper_params: Dict[str, Any] = {}, extra_params: Dict[str, Any] = {})

This class specifies an approximation function for a member of torch.nn.functional.

__init__(func_id: None, algorithm: str = 'vsimd', wrapper_params: Dict[str, Any] = {}, extra_params: Dict[str, Any] = {})

Methods

__init__(func_id[, algorithm, ...])
execute(*args, **kwargs)
from_shorthand(sh)