python - 是否有一种单行方式可以从张量流数据管道中获取下一个 N 个样本，而无需使用 for 循环？

Question

下面是我目前使用的，但更喜欢在没有 for 循环的情况下获得批量张量，这样我就可以一次性对整个批次进行预测，而不是循环遍历每个实例并能够将预测与基本事实 y_true 进行比较

for i, mini_batch in enumerate(val):
        x_val = np.array(mini_batch[0]['input_1'])
        y_true = np.array(mini_batch[1]['conv2d_12'])
        y_pred = autoencoder.predict(x_val)

标签： pythontensorflowmachine-learningtensorflow2.0tensorflow-datasets