deeplima::nets::torch_modules::DeepBiaffineAttentionLabelDecoderImpl Class Reference

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

Inheritance diagram for deeplima::nets::torch_modules::DeepBiaffineAttentionLabelDecoderImpl:

Public Member Functions
	DeepBiaffineAttentionLabelDecoderImpl (int64_t input_dim, int64_t hidden_dim, int64_t num_labels, bool input_includes_root=false)
int64_t	num_labels () const
bool	input_includes_root () const
torch::Tensor	forward (torch::Tensor input)

Public Attributes
bool	m_input_includes_root
int64_t	m_hidden_dim
int64_t	m_num_labels
torch::nn::Linear	mlp_head
torch::nn::Linear	mlp_dep
torch::nn::ELU	elu
torch::Tensor	U
torch::Tensor	root

Detailed Description

Definition at line 33 of file deep_biaffine_attention_label_decoder.h.

Constructor & Destructor Documentation

DeepBiaffineAttentionLabelDecoderImpl()

deeplima::nets::torch_modules::DeepBiaffineAttentionLabelDecoderImpl::DeepBiaffineAttentionLabelDecoderImpl ( int64_t  input_dim, int64_t  hidden_dim, int64_t  num_labels, bool  input_includes_root = false  )

inline

Definition at line 36 of file deep_biaffine_attention_label_decoder.h.

Member Function Documentation

forward()

torch::Tensor deeplima::nets::torch_modules::DeepBiaffineAttentionLabelDecoderImpl::forward

(

torch::Tensor

input

)

Definition at line 15 of file deep_biaffine_attention_label_decoder.cpp.

input_includes_root()

bool deeplima::nets::torch_modules::DeepBiaffineAttentionLabelDecoderImpl::input_includes_root ( ) const

inline

Definition at line 55 of file deep_biaffine_attention_label_decoder.h.

num_labels()

int64_t deeplima::nets::torch_modules::DeepBiaffineAttentionLabelDecoderImpl::num_labels ( ) const

inline

Definition at line 54 of file deep_biaffine_attention_label_decoder.h.

Member Data Documentation

elu

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

Definition at line 66 of file deep_biaffine_attention_label_decoder.h.

m_hidden_dim

int64_t deeplima::nets::torch_modules::DeepBiaffineAttentionLabelDecoderImpl::m_hidden_dim

Definition at line 62 of file deep_biaffine_attention_label_decoder.h.

m_input_includes_root

bool deeplima::nets::torch_modules::DeepBiaffineAttentionLabelDecoderImpl::m_input_includes_root

Definition at line 61 of file deep_biaffine_attention_label_decoder.h.

m_num_labels

int64_t deeplima::nets::torch_modules::DeepBiaffineAttentionLabelDecoderImpl::m_num_labels

Definition at line 63 of file deep_biaffine_attention_label_decoder.h.

mlp_dep

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

Definition at line 65 of file deep_biaffine_attention_label_decoder.h.

mlp_head

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

Definition at line 64 of file deep_biaffine_attention_label_decoder.h.

root

torch::Tensor deeplima::nets::torch_modules::DeepBiaffineAttentionLabelDecoderImpl::root

Definition at line 68 of file deep_biaffine_attention_label_decoder.h.

U

torch::Tensor deeplima::nets::torch_modules::DeepBiaffineAttentionLabelDecoderImpl::U

Definition at line 67 of file deep_biaffine_attention_label_decoder.h.

---

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

• deeplima/libs/nn/torch_modules/deep_biaffine_attention_label_decoder.h
• deeplima/libs/nn/torch_modules/deep_biaffine_attention_label_decoder.cpp