Belle II Software  release-08-01-10
Model Class Reference

Class for the ConvNet generator model. More...

Inheritance diagram for Model:
Collaboration diagram for Model:

Public Member Functions

def __init__ (self)
 Constructor to create a new model instance.
 
def forward (self, z)
 Function to perform a forward pass. More...
 

Public Attributes

 fc
 Fully-connected layer.
 
 features
 Sequential composite layer.
 

Detailed Description

Class for the ConvNet generator model.

ConvNet generator model.

Definition at line 18 of file convnet.py.

Member Function Documentation

◆ forward()

def 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.

60  def forward(self, z):
61  """Compute the model output for a given input."""
62  return self.features(self.fc(z).view(-1, 512, 8, 24)).tanh_()
63 

The documentation for this class was generated from the following file: