Skip to content
ml-recall

#batched-matmul

1 question from PyTorch

Timed test

1 question · 1 min · score and explanations at the end

Take a test
Q 1 / 1Score 0 / 0

Question 1: Short answer

hardAI-checkedPyTorch

What shape does this expression produce?

import torch
torch.matmul(torch.randn(4, 1, 3, 5), torch.randn(6, 5, 2)).shape

Four dimensions in order, e.g. (8, 1, 7, 9)