import numpy as np
x = np.ones(5)
print(x)
a = [3, 1, 2, 4, 6, 1]
print(np.argmax(a))

99ec9205e9c8a37e224aeafa8137db514ad2f86f
<mask token>

<mask token>
print(x)
<mask token>
print(np.argmax(a))

<mask token>
x = np.ones(5)
print(x)
a = [3, 1, 2, 4, 6, 1]
print(np.argmax(a))

import numpy as np
x = np.ones(5)
print(x)
a = [3, 1, 2, 4, 6, 1]
print(np.argmax(a))
