deeplima::nets::torch_modules::StanzaDepparseParserImpl Class Reference

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

Inheritance diagram for deeplima::nets::torch_modules::StanzaDepparseParserImpl:

Public Member Functions
	StanzaDepparseParserImpl ()=default
	StanzaDepparseParserImpl (int64_t word_emb_dim, int64_t tag_emb_dim, int64_t hidden_dim, int64_t num_layers, float dropout, float rec_dropout, std::shared_ptr< std::map< std::string, std::vector< std::string > > > vocab, std::shared_ptr< std::vector< std::vector< std::string > > > feats_vocabs, int64_t deep_biaff_hidden_dim, bool linearize, bool dist, int64_t word_dropout)
std::pair< float, std::vector< torch::Tensor > >	forward (torch::Tensor word, torch::Tensor word_mask, torch::Tensor wordchars, torch::Tensor wordchars_mask, torch::Tensor upos, torch::Tensor xpos, torch::Tensor ufeats, torch::Tensor pretrained, torch::Tensor lemma, torch::Tensor head, torch::Tensor deprels, torch::Tensor word_orig_idx, torch::Tensor sentlens, torch::Tensor wordlens)

Detailed Description

Definition at line 535 of file stanza_models_depparse_model.h.

Constructor & Destructor Documentation

StanzaDepparseParserImpl() [1/2]

deeplima::nets::torch_modules::StanzaDepparseParserImpl::StanzaDepparseParserImpl ( )

default

StanzaDepparseParserImpl() [2/2]

deeplima::nets::torch_modules::StanzaDepparseParserImpl::StanzaDepparseParserImpl	(	int64_t	word_emb_dim,
		int64_t	tag_emb_dim,
		int64_t	hidden_dim,
		int64_t	num_layers,
		float	dropout,
		float	rec_dropout,
		std::shared_ptr< std::map< std::string, std::vector< std::string > > >	vocab,
		std::shared_ptr< std::vector< std::vector< std::string > > >	feats_vocabs,
		int64_t	deep_biaff_hidden_dim,
		bool	linearize,
		bool	dist,
		int64_t	word_dropout
	)

Definition at line 27 of file stanza_models_depparse_model.cpp.

Member Function Documentation

forward()

std::pair< float, std::vector< torch::Tensor > > deeplima::nets::torch_modules::StanzaDepparseParserImpl::forward	(	torch::Tensor	word,
		torch::Tensor	word_mask,
		torch::Tensor	wordchars,
		torch::Tensor	wordchars_mask,
		torch::Tensor	upos,
		torch::Tensor	xpos,
		torch::Tensor	ufeats,
		torch::Tensor	pretrained,
		torch::Tensor	lemma,
		torch::Tensor	head,
		torch::Tensor	deprels,
		torch::Tensor	word_orig_idx,
		torch::Tensor	sentlens,
		torch::Tensor	wordlens
	)

Definition at line 120 of file stanza_models_depparse_model.cpp.

---

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

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