LIMA
Libre Multilingual Analyzer — C++ API
Loading...
Searching...
No Matches
Public Member Functions | Public Attributes | List of all members
deeplima::nets::torch_modules::PairwiseBilinearImpl Class Reference

A bilinear module that deals with broadcasting for efficient memory usage. More...

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

Inheritance diagram for deeplima::nets::torch_modules::PairwiseBilinearImpl:

Public Member Functions

 PairwiseBilinearImpl ()=default
 
 PairwiseBilinearImpl (int64_t input1_size, int64_t input2_size, int output_size, bool bias=true)
 
torch::Tensor forward (torch::Tensor input1, torch::Tensor input2)
 

Public Attributes

int64_t input1_size
 
int64_t input2_size
 
int64_t output_size
 

Detailed Description

A bilinear module that deals with broadcasting for efficient memory usage.

Input: tensors of sizes (N x L1 x D1) and (N x L2 x D2) Output: tensor of size (N x L1 x L2 x O)

Definition at line 411 of file stanza_models_depparse_model.h.

Constructor & Destructor Documentation

◆ PairwiseBilinearImpl() [1/2]

deeplima::nets::torch_modules::PairwiseBilinearImpl::PairwiseBilinearImpl ( )
default

◆ PairwiseBilinearImpl() [2/2]

deeplima::nets::torch_modules::PairwiseBilinearImpl::PairwiseBilinearImpl ( int64_t  input1_size,
int64_t  input2_size,
int  output_size,
bool  bias = true 
)
inline

Definition at line 415 of file stanza_models_depparse_model.h.

Member Function Documentation

◆ forward()

torch::Tensor deeplima::nets::torch_modules::PairwiseBilinearImpl::forward ( torch::Tensor  input1,
torch::Tensor  input2 
)
inline

Definition at line 427 of file stanza_models_depparse_model.h.

Member Data Documentation

◆ input1_size

int64_t deeplima::nets::torch_modules::PairwiseBilinearImpl::input1_size

Definition at line 446 of file stanza_models_depparse_model.h.

◆ input2_size

int64_t deeplima::nets::torch_modules::PairwiseBilinearImpl::input2_size

Definition at line 446 of file stanza_models_depparse_model.h.

◆ output_size

int64_t deeplima::nets::torch_modules::PairwiseBilinearImpl::output_size

Definition at line 447 of file stanza_models_depparse_model.h.

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