3 / 6 NumPy & JAX

Keepdims Shape

For x with shape (4, 5, 6), what is the shape of x.sum(axis=1, keepdims=True)?

A

(4, 5, 1)

B

(4, 6, 1)

C

(4, 6)

D

(4, 1, 6)

E

(1, 5, 6)