spyrit.core.nnet.ConvNet.forward

ConvNet.forward(x: tensor) → tensor

Forward pass of the ConvNet model.

Args:

x (torch.tensor): The input tensor. It is expected to have the shape b, 1, h, w, where b is the batch size, h is the height, and w is the width.

Returns:

torch.tensor: The output tensor of the ConvNet model. It has shape b, 1, h, w.