#include </home/runner/work/lima/lima/deeplima/libs/nn/torch_modules/stanza_models_depparse_model.h>
|
| | PackedLSTMImpl ()=default |
| |
| | PackedLSTMImpl (int64_t input_size, int64_t hidden_size, int64_t num_layers, bool bias=true, bool batch_first=false, float dropout=0, bool bidirectional=false, bool pad=false, float rec_dropout=0) |
| |
| std::tuple< torch::Tensor, std::tuple< torch::Tensor, torch::Tensor > > | forward (torch::Tensor input, torch::Tensor lengths, torch::optional< std::tuple< torch::Tensor, torch::Tensor > > hx={}) |
| |
| std::tuple< torch::nn::utils::rnn::PackedSequence, std::tuple< torch::Tensor, torch::Tensor > > | forward_with_packed_input (const torch::nn::utils::rnn::PackedSequence &packed_input, torch::Tensor lengths, torch::optional< std::tuple< torch::Tensor, torch::Tensor > > hx={}) |
| |
◆ PackedLSTMImpl() [1/2]
| deeplima::nets::torch_modules::PackedLSTMImpl::PackedLSTMImpl |
( |
| ) |
|
|
default |
◆ PackedLSTMImpl() [2/2]
| deeplima::nets::torch_modules::PackedLSTMImpl::PackedLSTMImpl |
( |
int64_t |
input_size, |
|
|
int64_t |
hidden_size, |
|
|
int64_t |
num_layers, |
|
|
bool |
bias = true, |
|
|
bool |
batch_first = false, |
|
|
float |
dropout = 0, |
|
|
bool |
bidirectional = false, |
|
|
bool |
pad = false, |
|
|
float |
rec_dropout = 0 |
|
) |
| |
|
inline |
◆ forward()
| std::tuple< torch::Tensor, std::tuple< torch::Tensor, torch::Tensor > > deeplima::nets::torch_modules::PackedLSTMImpl::forward |
( |
torch::Tensor |
input, |
|
|
torch::Tensor |
lengths, |
|
|
torch::optional< std::tuple< torch::Tensor, torch::Tensor > > |
hx = {} |
|
) |
| |
|
inline |
◆ forward_with_packed_input()
| std::tuple< torch::nn::utils::rnn::PackedSequence, std::tuple< torch::Tensor, torch::Tensor > > deeplima::nets::torch_modules::PackedLSTMImpl::forward_with_packed_input |
( |
const torch::nn::utils::rnn::PackedSequence & |
packed_input, |
|
|
torch::Tensor |
lengths, |
|
|
torch::optional< std::tuple< torch::Tensor, torch::Tensor > > |
hx = {} |
|
) |
| |
|
inline |
◆ batch_first
| bool deeplima::nets::torch_modules::PackedLSTMImpl::batch_first |
◆ lstm
| torch::nn::AnyModule deeplima::nets::torch_modules::PackedLSTMImpl::lstm |
◆ pad
| bool deeplima::nets::torch_modules::PackedLSTMImpl::pad |
The documentation for this class was generated from the following file:
1.9.8