import numpy as np
x = np.array([[0, 1, 2], [3, 4, 5], [5, 6, 7], [7, 8, 9]])
print(x)
print(x[..., [0, 2]])
print(x[2, ...])
print(x[[1, 2], ...])

a050bd8044da4acd19118aee34bdd6cd5a25a231
<mask token>

<mask token>
print(x)
print(x[..., [0, 2]])
print(x[2, ...])
print(x[[1, 2], ...])

<mask token>
x = np.array([[0, 1, 2], [3, 4, 5], [5, 6, 7], [7, 8, 9]])
print(x)
print(x[..., [0, 2]])
print(x[2, ...])
print(x[[1, 2], ...])

import numpy as np
x = np.array([[0, 1, 2], [3, 4, 5], [5, 6, 7], [7, 8, 9]])
print(x)
print(x[..., [0, 2]])
print(x[2, ...])
print(x[[1, 2], ...])
