https://indianaiproduction.com/python-numpy-shape/