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

Inheritance diagram for deeplima::nets::torch_modules::DeepBiaffineAttentionDecoderImpl:

Public Member Functions
	DeepBiaffineAttentionDecoderImpl (int64_t input_dim, int64_t hidden_arc_dim, bool input_includes_root=false)

torch::Tensor	forward (torch::Tensor input)

Public Attributes
bool	m_input_includes_root

int64_t	m_hidden_arc_dim

torch::nn::Linear	mlp_head

torch::nn::Linear	mlp_dep

torch::nn::ELU	elu

torch::Tensor	U1

torch::Tensor	u2

torch::Tensor	root

torch::Tensor	root2

Detailed Description

Definition at line 18 of file deep_biaffine_attention_decoder.h.

Constructor & Destructor Documentation

◆ DeepBiaffineAttentionDecoderImpl()

deeplima::nets::torch_modules::DeepBiaffineAttentionDecoderImpl::DeepBiaffineAttentionDecoderImpl	(	int64_t	input_dim,
		int64_t	hidden_arc_dim,
		bool	input_includes_root = `false`
	)

inline

Definition at line 21 of file deep_biaffine_attention_decoder.h.

Member Function Documentation

◆ forward()

torch::Tensor deeplima::nets::torch_modules::DeepBiaffineAttentionDecoderImpl::forward ( torch::Tensor input )

Definition at line 15 of file deep_biaffine_attention_decoder.cpp.

Member Data Documentation

◆ elu

torch::nn::ELU deeplima::nets::torch_modules::DeepBiaffineAttentionDecoderImpl::elu

Definition at line 41 of file deep_biaffine_attention_decoder.h.

◆ m_hidden_arc_dim

int64_t deeplima::nets::torch_modules::DeepBiaffineAttentionDecoderImpl::m_hidden_arc_dim

Definition at line 38 of file deep_biaffine_attention_decoder.h.

◆ m_input_includes_root

bool deeplima::nets::torch_modules::DeepBiaffineAttentionDecoderImpl::m_input_includes_root

Definition at line 37 of file deep_biaffine_attention_decoder.h.

◆ mlp_dep

torch::nn::Linear deeplima::nets::torch_modules::DeepBiaffineAttentionDecoderImpl::mlp_dep

Definition at line 40 of file deep_biaffine_attention_decoder.h.

◆ mlp_head

torch::nn::Linear deeplima::nets::torch_modules::DeepBiaffineAttentionDecoderImpl::mlp_head

Definition at line 39 of file deep_biaffine_attention_decoder.h.

◆ root

torch::Tensor deeplima::nets::torch_modules::DeepBiaffineAttentionDecoderImpl::root

Definition at line 43 of file deep_biaffine_attention_decoder.h.

◆ root2

torch::Tensor deeplima::nets::torch_modules::DeepBiaffineAttentionDecoderImpl::root2

Definition at line 43 of file deep_biaffine_attention_decoder.h.

◆ U1

torch::Tensor deeplima::nets::torch_modules::DeepBiaffineAttentionDecoderImpl::U1

Definition at line 42 of file deep_biaffine_attention_decoder.h.

◆ u2

torch::Tensor deeplima::nets::torch_modules::DeepBiaffineAttentionDecoderImpl::u2