#keepdims

Question 1: Single choice

mediumAI-checkedPyTorch

What happens when the last line runs?

A = torch.arange(6, dtype=torch.float32).reshape(2, 3)
A / A.sum(axis=1)

Each column of A is divided by its own sum, so every column of the result sums to 11
The whole matrix is divided by the sum of all its entries, which is 152
A RuntimeError: the (2,) row sums are lined up with the last axis, which has size 33
Each row of A is divided by its own sum, so that every row of the result sums to exactly 14