torch.nn.functional.scaled\_mm ============================== .. currentmodule:: torch.nn.functional .. autofunction:: scaled_mm