ml-recall

#custom-layer

1 question from PyTorch

Timed test

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

Take a test

Q 1 / 1Score 0 / 0

Shuffle questions

Question 1: Single choice

mediumAI-checkedPyTorch

A custom layer computes $y_k = \sum_{i,j} W_{ijk}\, x_i x_j$ for every example, with a parameter W of shape (d, d, m). For a batch X of shape (n, d), which line returns the (n, m) output?

torch.einsum("ni,nj,ijk->nk", X, X, W)1
torch.einsum("ni,nj,ijk->k", X, X, W)2
torch.einsum("ni,ni,iik->nk", X, X, W)3
torch.einsum("ni,ijk->nk", X, W)4

Keys: 1–9 select · Enter check / next