Python 将4d阵列存储为tfrec或作为批处理检索它们

好吧，我不太习惯使用TensorFlow，但我在使用TPU时遇到了一个恼人的问题

我有以下资料：

一个4D数组（一个带有空间信息的图像），比如np.ones（（5,5,5,3）），我想变成一个TFR记录

现在我有两种方法可以解决这个问题，如果我能解决任何一种，我都会很好。我将5个图像5x（5,5,3）保存到一个.tfrec文件中。但现在我不知道如何批量打开一个文件。不幸的是，所有数组的第一维都不同，有些是（3,5,5,3），有些可能是（8,5,5,3），所以我不能将第一维设置为批量大小

有没有办法将tfrec文件指定为单个示例？或者这是否太低效了，因为我会为每个4D图像使用不同的.tfrec文件

另一种方法是将整个4D数组保存为字符串？我不知道该怎么办。到目前为止，我使用了cv2.imencode，但它只适用于3d阵列

我很感谢你的建议