aibedo.models.BaseModel.forward

BaseModel.forward(X: torch.Tensor)

Standard ML model forward pass (to be implemented by the specific ML model).

Parameters

X (Tensor) – Input data tensor of shape \((B, *, C_{in})\)

Shapes:

• Input: \((B, *, C_{in})\),

where \(B\) is the batch size, \(*\) is the spatial dimension(s) of the data, and \(C_{in}\) is the number of input features/channels.