Ones

import numpy as np

Numpy Ones

a = np.ones(5)
print(a)
[1. 1. 1. 1. 1.]

b = np.ones([2, 2], dtype=np.float16)
print(b)
[[1. 1.]
 [1. 1.]]