spyrit.core.nnet.ConvNetBN.forward

ConvNetBN.forward(x: tensor) → tensor

Forward pass of the ConvNetBN model.

Args:

x (torch.tensor): 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 ConvNetBN model. It has shape b, 1, h, w.