MNISTData.infer_on_batch#

MNISTData.infer_on_batch(batch, model, device) Union[List[Tensor], Tensor][source]#

Infer on a batch of mnist data.