florist.api.models.mnist module¶

Definitions for the MNIST model.

class MnistNet[source]¶

Bases: Module

Implementation of the Mnist model.

__init__()[source]¶: Initialize an instance of MnistNet.

forward(x)[source]¶

Perform a forward pass for the given tensor.

Parameters:: x (Tensor) – (torch.Tensor) the tensor to perform the forward pass on.
Return type:: Tensor
Returns:: (torch.Tensor) a result tensor after the forward pass.