cnn

Base CNN-specific Model class(es).

class graphnet.models.cnn.cnn.CNN(*args, **kwargs)

Bases: Model

Base class for all core CNN models in graphnet.

Construct CNN.

Parameters:

• args (Any)

• kwargs (Any)

Return type:

object

property nb_inputs: int

Return number of input features.

property nb_outputs: int

Return number of output features.

abstractmethod forward(data)

Apply learnable forward pass in model.

Return type:

Tensor

Parameters:

data (Data)