caffe2.python.models.seq2seq.seq2seq_util Namespace Reference

Classes
class	LSTMWithAttentionDecoder

Functions
def	gen_vocab (corpus, unk_threshold)
def	get_numberized_sentence (sentence, vocab)
def	rnn_unidirectional_layer (model, inputs, input_lengths, input_size, num_units, dropout_keep_prob, forward_only, return_sequence_output, return_final_state, scope=None)
def	rnn_bidirectional_layer (model, inputs, input_lengths, input_size, num_units, dropout_keep_prob, forward_only, return_sequence_output, return_final_state, scope=None)
def	build_embeddings (model, vocab_size, embedding_size, name, freeze_embeddings)
def	get_layer_scope (scope, layer_type, i)
def	build_embedding_encoder (model, encoder_params, num_decoder_layers, inputs, input_lengths, vocab_size, embeddings, embedding_size, use_attention, num_gpus=0, forward_only=False, scope=None)
def	build_initial_rnn_decoder_states (model, encoder_units_per_layer, decoder_units_per_layer, final_encoder_hidden_states, final_encoder_cell_states, use_attention)
def	build_embedding_decoder (model, decoder_layer_configs, inputs, input_lengths, encoder_lengths, encoder_outputs, weighted_encoder_outputs, final_encoder_hidden_states, final_encoder_cell_states, encoder_units_per_layer, vocab_size, embeddings, embedding_size, attention_type, forward_only, num_gpus=0, scope=None)
def	output_projection (model, decoder_outputs, decoder_output_size, target_vocab_size, decoder_softmax_size)

Variables
int	PAD_ID = 0
string	PAD = '<PAD>'
int	GO_ID = 1
string	GO = '<GO>'
int	EOS_ID = 2
string	EOS = '<EOS>'
int	UNK_ID = 3
string	UNK = '<UNK>'

Function Documentation

build_embedding_decoder()

def caffe2.python.models.seq2seq.seq2seq_util.build_embedding_decoder	(		model,
			decoder_layer_configs,
			inputs,
			input_lengths,
			encoder_lengths,
			encoder_outputs,
			weighted_encoder_outputs,
			final_encoder_hidden_states,
			final_encoder_cell_states,
			encoder_units_per_layer,
			vocab_size,
			embeddings,
			embedding_size,
			attention_type,
			forward_only,
			num_gpus = 0,
			scope = None
	)

Definition at line 519 of file seq2seq_util.py.

References caffe2.python.models.seq2seq.seq2seq_util.build_embedding_decoder(), caffe2.python.models.seq2seq.seq2seq_util.build_initial_rnn_decoder_states(), c10::prim.enumerate(), and caffe2.python.models.seq2seq.seq2seq_util.get_layer_scope().

Referenced by caffe2.python.models.seq2seq.seq2seq_util.build_embedding_decoder().

build_embedding_encoder()

def caffe2.python.models.seq2seq.seq2seq_util.build_embedding_encoder	(		model,
			encoder_params,
			num_decoder_layers,
			inputs,
			input_lengths,
			vocab_size,
			embeddings,
			embedding_size,
			use_attention,
			num_gpus = 0,
			forward_only = False,
			scope = None
	)

Definition at line 224 of file seq2seq_util.py.

References caffe2.python.models.seq2seq.seq2seq_util.build_embedding_encoder(), c10::prim.enumerate(), caffe2.python.models.seq2seq.seq2seq_util.get_layer_scope(), and c10::aten.len().

Referenced by caffe2.python.models.seq2seq.seq2seq_util.build_embedding_encoder().

build_embeddings()

def caffe2.python.models.seq2seq.seq2seq_util.build_embeddings	(		model,
			vocab_size,
			embedding_size,
			name,
			freeze_embeddings
	)

Definition at line 201 of file seq2seq_util.py.

References caffe2.python.models.seq2seq.seq2seq_util.build_embeddings().

Referenced by caffe2.python.models.seq2seq.seq2seq_util.build_embeddings().

build_initial_rnn_decoder_states()

def caffe2.python.models.seq2seq.seq2seq_util.build_initial_rnn_decoder_states	(		model,
			encoder_units_per_layer,
			decoder_units_per_layer,
			final_encoder_hidden_states,
			final_encoder_cell_states,
			use_attention
	)

Definition at line 428 of file seq2seq_util.py.

References caffe2.python.models.seq2seq.seq2seq_util.build_initial_rnn_decoder_states(), c10::prim.enumerate(), caffe2.format(), and c10::aten.len().

Referenced by caffe2.python.models.seq2seq.seq2seq_util.build_embedding_decoder(), and caffe2.python.models.seq2seq.seq2seq_util.build_initial_rnn_decoder_states().

gen_vocab()

def caffe2.python.models.seq2seq.seq2seq_util.gen_vocab	(		corpus,
			unk_threshold
	)

Definition at line 27 of file seq2seq_util.py.

References caffe2.python.models.seq2seq.seq2seq_util.gen_vocab(), and c10::aten.len().

Referenced by caffe2.python.models.seq2seq.seq2seq_util.gen_vocab().

get_layer_scope()

def caffe2.python.models.seq2seq.seq2seq_util.get_layer_scope	(		scope,
			layer_type,
			i
	)

Definition at line 219 of file seq2seq_util.py.

References caffe2.format(), and caffe2.python.models.seq2seq.seq2seq_util.get_layer_scope().

Referenced by caffe2.python.models.seq2seq.seq2seq_util.build_embedding_decoder(), caffe2.python.models.seq2seq.seq2seq_util.build_embedding_encoder(), and caffe2.python.models.seq2seq.seq2seq_util.get_layer_scope().

get_numberized_sentence()

def caffe2.python.models.seq2seq.seq2seq_util.get_numberized_sentence	(		sentence,
			vocab
	)

Definition at line 48 of file seq2seq_util.py.

References caffe2.python.models.seq2seq.seq2seq_util.get_numberized_sentence().

Referenced by caffe2.python.models.seq2seq.seq2seq_util.get_numberized_sentence().

output_projection()

def caffe2.python.models.seq2seq.seq2seq_util.output_projection	(		model,
			decoder_outputs,
			decoder_output_size,
			target_vocab_size,
			decoder_softmax_size
	)

Definition at line 632 of file seq2seq_util.py.

References caffe2.python.models.seq2seq.seq2seq_util.output_projection().

Referenced by caffe2.python.models.seq2seq.seq2seq_util.output_projection().

rnn_bidirectional_layer()

def caffe2.python.models.seq2seq.seq2seq_util.rnn_bidirectional_layer	(		model,
			inputs,
			input_lengths,
			input_size,
			num_units,
			dropout_keep_prob,
			forward_only,
			return_sequence_output,
			return_final_state,
			scope = None
	)

Definition at line 124 of file seq2seq_util.py.

References caffe2.python.models.seq2seq.seq2seq_util.rnn_bidirectional_layer(), and caffe2.python.models.seq2seq.seq2seq_util.rnn_unidirectional_layer().

Referenced by caffe2.python.models.seq2seq.seq2seq_util.rnn_bidirectional_layer().

rnn_unidirectional_layer()

def caffe2.python.models.seq2seq.seq2seq_util.rnn_unidirectional_layer	(		model,
			inputs,
			input_lengths,
			input_size,
			num_units,
			dropout_keep_prob,
			forward_only,
			return_sequence_output,
			return_final_state,
			scope = None
	)

 Unidirectional LSTM encoder.

Definition at line 58 of file seq2seq_util.py.

References caffe2.python.models.seq2seq.seq2seq_util.rnn_unidirectional_layer().

Referenced by caffe2.python.models.seq2seq.seq2seq_util.rnn_bidirectional_layer(), and caffe2.python.models.seq2seq.seq2seq_util.rnn_unidirectional_layer().

Variable Documentation

EOS

string caffe2.python.models.seq2seq.seq2seq_util.EOS = '<EOS>'

Definition at line 22 of file seq2seq_util.py.

EOS_ID

int caffe2.python.models.seq2seq.seq2seq_util.EOS_ID = 2

Definition at line 21 of file seq2seq_util.py.

GO

string caffe2.python.models.seq2seq.seq2seq_util.GO = '<GO>'

Definition at line 20 of file seq2seq_util.py.

Referenced by caffe2.Arg(), caffe2.argmaxes(), caffe2::FeedBlob::GetExpGradient.GetGradientDefs(), caffe2::GetFloatToHalfGradient.GetGradientDefs(), caffe2::GetHalfToFloatGradient.GetGradientDefs(), caffe2::GetLeakyReluGradient.GetGradientDefs(), caffe2::GetTTSparseLengthsGradient.GetGradientDefs(), caffe2::FeedBlob::GetLogGradient.GetGradientDefs(), caffe2::FeedBlob::GetSqrtGradient.GetGradientDefs(), caffe2.gradient(), caffe2.IdenticalTypeAndShapeOfInput(), caffe2.Input(), caffe2.MaskWithFunctor(), caffe2.NumOutputs(), caffe2.OPERATOR_SCHEMA(), caffe2.Output(), caffe2.REGISTER_GRADIENT(), and caffe2.SetDoc().

GO_ID

int caffe2.python.models.seq2seq.seq2seq_util.GO_ID = 1

Definition at line 19 of file seq2seq_util.py.

PAD

string caffe2.python.models.seq2seq.seq2seq_util.PAD = '<PAD>'

Definition at line 18 of file seq2seq_util.py.

PAD_ID

int caffe2.python.models.seq2seq.seq2seq_util.PAD_ID = 0

Definition at line 17 of file seq2seq_util.py.

UNK

string caffe2.python.models.seq2seq.seq2seq_util.UNK = '<UNK>'

Definition at line 24 of file seq2seq_util.py.

UNK_ID

int caffe2.python.models.seq2seq.seq2seq_util.UNK_ID = 3

Definition at line 23 of file seq2seq_util.py.