spyrit.core.nnet.DConvNet.forward

DConvNet.forward(x: tensor) → tensor

Forward pass of the DConvNet 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 DConvNet model. It has shape b, 1, h, w.