Rate this Page
★ ★ ★ ★ ★

torch.utils.flop_counter.shape_wrapper#

torch.utils.flop_counter.shape_wrapper(f)[source]#
On this page
Show Source
PyTorch Libraries