python 将3D数组转换为列向量

Posted 2021-05-09

def image2vector(image):
    """
    Argument:
    image -- a numpy array of shape (length, height, depth)
    
    Returns:
    v -- a vector of shape (length*height*depth, 1)
    """
    
    v = image.reshape((image.shape[0] * image.shape[1] * image.shape[2], 1))
    
    return v