Rate this Page
★ ★ ★ ★ ★

torch.utils.flop_counter.register_flop_formula#

torch.utils.flop_counter.register_flop_formula(targets, get_raw=False)[source]#
Return type:

Callable[[Callable[[~_P], _T]], Callable[[~_P], _T]]

On this page
Show Source
PyTorch Libraries