We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
torch.matmul
mm
[S, 1, H] @ [H, K]