medium10 pts · first answer countspytorchtensor-shapes

Shape after a batched matmul

What does this code print?

import torch
a = torch.randn(8, 10, 64)
w = torch.randn(64, 32)
print((a @ w).shape)

A.torch.Size([8, 10, 32])B.torch.Size([8, 64, 32])C.torch.Size([10, 32])D.RuntimeError: shape mismatch

Discussion 0

Join free to post in the discussion.