TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends Pages
layers.LinearNorm Class Reference
Inheritance diagram for layers.LinearNorm:
Collaboration diagram for layers.LinearNorm:

Public Member Functions

def __init__ (self, in_dim, out_dim, bias=True, w_init_gain='linear')
 
def forward (self, x)
 

Public Attributes

 linear_layer
 

Constructor & Destructor Documentation

◆ __init__()

def layers.LinearNorm.__init__ (   self,
  in_dim,
  out_dim,
  bias = True,
  w_init_gain = 'linear' 
)

Member Function Documentation

◆ forward()

def layers.LinearNorm.forward (   self,
  x 
)

Member Data Documentation

◆ linear_layer

layers.LinearNorm.linear_layer

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