predict_on_batch

FullyConnected.predict_on_batch(x)

Returns predictions for a single batch of samples.

Parameters:

x –

Input data. It could be: - A Numpy array (or array-like), or a list of arrays (in case the

model has multiple inputs).

• A TensorFlow tensor, or a list of tensors (in case the model has

  multiple inputs).

Returns:

Numpy array(s) of predictions.

Raises:

RuntimeError – If model.predict_on_batch is wrapped in a tf.function.