![]() |
Belle II Software development
|
Class for the ConvNet generator model. More...
Public Member Functions | |
__init__ (self) | |
Constructor to create a new model instance. | |
forward (self, z) | |
Function to perform a forward pass. | |
Public Attributes | |
fc = nn.Linear(96, 98304) | |
Fully-connected layer. | |
features | |
Sequential composite layer. | |
Class for the ConvNet generator model.
ConvNet generator model.
Definition at line 18 of file convnet.py.
__init__ | ( | self | ) |
Constructor to create a new model instance.
Definition at line 23 of file convnet.py.
forward | ( | self, | |
z ) |
Function to perform a forward pass.
Compute the model output for a given input.
Definition at line 60 of file convnet.py.
fc = nn.Linear(96, 98304) |
Fully-connected layer.
Definition at line 25 of file convnet.py.
features |
Sequential composite layer.
Definition at line 26 of file convnet.py.