Есть ли функция

import numpy as np
 
heigth = 600
width = 400
 
input = np.zeros((heigth * width, 2))
 
step = -1
 
for i in range(heigth):
    for j in range(width):        
        step = step + 1        
        input[step,0] = i
        input[step,1] = j

import numpy as np
 
heigth = 600
width = 400
b,d=np.meshgrid(np.arange(width), np.arange(heigth))
result=np.dstack((d,b)).reshape(-1,2)

result = np.dstack(np.meshgrid(np.arange(width), np.arange(heigth))[::-1]).reshape(-1,2)

import tensorflow as tf
tf.reshape(tf.experimental.numpy.dstack(tf.meshgrid(tf.experimental.numpy.arange(width), tf.experimental.numpy.arange(heigth))[::-1]), (-1,2))

import numpy as np
 
heigth = 600
width = 400
 
img_origin=np.random.randint(0,255, (heigth, width, 3), np.uint8)
 
y = np.zeros((width * heigth, 3))
 
step = -1;
 
for i in range(heigth):
    for j in range(width):        
        step = step + 1        
        for k in range(3):            
            y[step,k] = img_origin[i,j,k]

b, d, c = np.meshgrid(np.arange(width), np.arange(heigth), np.arange(3))
result=np.dstack((c, d, b)).reshape(-1,3)

import numpy as np
 
heigth = 600
width = 400
 
img_origin=np.random.randint(0,255, (heigth, width, 3), np.uint8)
 
y = np.zeros((width * heigth, 3))
 
step = -1
 
for i in range(heigth):
    for j in range(width):        
        step = step + 1        
        for k in range(3):            
            y[step,k] = img_origin[i,j,k]
            
#########################################
y1 = img_origin.reshape(heigth*width, 3)
print((y==y1).all())