Computes predictions for `input_batch`. The output can be in the form of probablistic distributions, e.g., softmax logits for discrete outputs, mixture of logistics for continuous values, or regression values. For training/evaluation, the output will be used for computing loss and
(self, theta, input_batch)
source not stored for this graph (policy: none)