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
model.Encoder Class Reference
Inheritance diagram for model.Encoder:
Collaboration diagram for model.Encoder:

Public Member Functions

def __init__ (self, encoder_n_convolutions, encoder_embedding_dim, encoder_kernel_size)
 
def forward (self, x, input_lengths)
 
def infer (self, x, input_lengths)
 

Public Attributes

 convolutions
 
 lstm
 

Detailed Description

Encoder module:
    - Three 1-d convolution banks
    - Bidirectional LSTM

Constructor & Destructor Documentation

◆ __init__()

def model.Encoder.__init__ (   self,
  encoder_n_convolutions,
  encoder_embedding_dim,
  encoder_kernel_size 
)

Member Function Documentation

◆ forward()

def model.Encoder.forward (   self,
  x,
  input_lengths 
)

◆ infer()

def model.Encoder.infer (   self,
  x,
  input_lengths 
)

Member Data Documentation

◆ convolutions

model.Encoder.convolutions

◆ lstm

model.Encoder.lstm

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