seq2seq_util Namespace Reference

Module caffe2.python.examples.seq2seq_util. More...

Classes
class	ModelHelper

Functions
def	rnn_unidirectional_encoder (model, embedded_inputs, input_lengths, initial_hidden_state, initial_cell_state, embedding_size, encoder_num_units, use_attention)
def	rnn_bidirectional_encoder (model, embedded_inputs, input_lengths, initial_hidden_state, initial_cell_state, embedding_size, encoder_num_units, use_attention)

Detailed Description

Module caffe2.python.examples.seq2seq_util.

Function Documentation

rnn_bidirectional_encoder()

def seq2seq_util.rnn_bidirectional_encoder	(		model,
			embedded_inputs,
			input_lengths,
			initial_hidden_state,
			initial_cell_state,
			embedding_size,
			encoder_num_units,
			use_attention
	)

Bidirectional (forward pass and backward pass) LSTM encoder.

Definition at line 91 of file seq2seq_util.py.

rnn_unidirectional_encoder()

def seq2seq_util.rnn_unidirectional_encoder	(		model,
			embedded_inputs,
			input_lengths,
			initial_hidden_state,
			initial_cell_state,
			embedding_size,
			encoder_num_units,
			use_attention
	)

Unidirectional (forward pass) LSTM encoder.

Definition at line 66 of file seq2seq_util.py.