In [8]:
# Creating a ones vector using ones function

a = np.ones([7])
b = np.ones([3,3])

print("a:", a)
print("b:", b)
a: [1. 1. 1. 1. 1. 1. 1.]
b: [[1. 1. 1.]
 [1. 1. 1.]
 [1. 1. 1.]]