构建用于采样图像的部分图像数据集

Question

我有一批形状为灰度图像（批大小，图像宽度，图像高度1）。由此，我想为该批次中的每个图像构建一个随机采样的像素值数组以及这些像素的相应笛卡尔位置的数组。

这些数组将被馈送到ML模型，该模型试图从有限数量的像素中重建图像。关键是输入的数据采用离散像素的形式，而不是不完整的图像。

我所有的数据都存储在numpy数组中。

我习惯于沿一个轴采样，但是我真的很困惑如何处理所有这些轴并得到正确形状的东西。