我面临的问题是我试图将一个函数应用于 keras 模型的模型输出中的每个样本，而不知道批量大小是多少。
为了进一步阐述这一点，我正在创建一个模型，我将在该模型上使用 predict_on_batch。假设模型看起来像 model = tf.keras.Model(x,y)。y = (batch_size, output_shape) 的形状，我正在尝试将函数应用于 batch_size 中每个样本的输出，并将其作为模型输出返回
我面临的问题是我试图将一个函数应用于 keras 模型的模型输出中的每个样本，而不知道批量大小是多少。
为了进一步阐述这一点，我正在创建一个模型，我将在该模型上使用 predict_on_batch。假设模型看起来像 model = tf.keras.Model(x,y)。y = (batch_size, output_shape) 的形状，我正在尝试将函数应用于 batch_size 中每个样本的输出，并将其作为模型输出返回