Highway LSTM network, does NOT use the HLSTMCell above A Highway LSTM network, as used in the original Tensorflow version of the Dozat parser. More...

#include </home/runner/work/lima/lima/deeplima/libs/nn/torch_modules/stanza_models_depparse_model.h>

Inheritance diagram for deeplima::nets::torch_modules::HighwayLSTMImpl:

Public Member Functions
	HighwayLSTMImpl ()=default

	HighwayLSTMImpl (int64_t input_size, int64_t hidden_size, int64_t num_layers=1, bool bias=true, bool batch_first=false, float dropout=0, bool bidirectional=false, float rec_dropout=0, std::function< torch::Tensor(const torch::Tensor &)> highway_func=nullptr, bool pad=false)

std::tuple< torch::Tensor, std::tuple< torch::Tensor, torch::Tensor > >	forward (torch::Tensor input, torch::Tensor seqlens, std::tuple< torch::Tensor, torch::Tensor > hx={torch::Tensor(), torch::Tensor()})

Public Attributes
int64_t	input_size

int64_t	hidden_size

int64_t	num_layers

bool	bias

bool	batch_first

float	dropout

bool	bidirectional

int64_t	num_directions

std::function< torch::Tensor(const torch::Tensor &)>	highway_func

bool	pad

torch::nn::ModuleList	lstm

torch::nn::ModuleList	highway

torch::nn::ModuleList	gate

torch::nn::Dropout	drop

Detailed Description

Highway LSTM network, does NOT use the HLSTMCell above A Highway LSTM network, as used in the original Tensorflow version of the Dozat parser.

Note that this is independent from the HLSTMCell above.

Definition at line 305 of file stanza_models_depparse_model.h.

Constructor & Destructor Documentation

◆ HighwayLSTMImpl() [1/2]

deeplima::nets::torch_modules::HighwayLSTMImpl::HighwayLSTMImpl ( )

default

◆ HighwayLSTMImpl() [2/2]

deeplima::nets::torch_modules::HighwayLSTMImpl::HighwayLSTMImpl	(	int64_t	input_size,
		int64_t	hidden_size,
		int64_t	num_layers = `1`,
		bool	bias = `true`,
		bool	batch_first = `false`,
		float	dropout = `0`,
		bool	bidirectional = `false`,
		float	rec_dropout = `0`,
		std::function< torch::Tensor(const torch::Tensor &)>	highway_func = `nullptr`,
		bool	pad = `false`
	)

inline

Definition at line 309 of file stanza_models_depparse_model.h.

Member Function Documentation

◆ forward()

std::tuple< torch::Tensor, std::tuple< torch::Tensor, torch::Tensor > > deeplima::nets::torch_modules::HighwayLSTMImpl::forward	(	torch::Tensor	input,
		torch::Tensor	seqlens,
		std::tuple< torch::Tensor, torch::Tensor >	hx = `{torch::Tensor(),torch::Tensor()}`
	)