Struct syntaxdot_transformers::layers::PairwiseBilinear

pub struct PairwiseBilinear { /* private fields */ }

Pairwise bilinear forms.

Given two batches with sequence length n, apply pairwise bilinear forms to each timestep within a sequence.

Implementations

impl PairwiseBilinear

pub fn new<'a>(
    vs: impl Borrow<PathExt<'a>>,
    config: &PairwiseBilinearConfig
) -> Result<Self, TransformerError>

Construct a new bilinear layer.

pub fn forward(&self, u: &Tensor, v: &Tensor) -> Result<Tensor, TransformerError>

Apply this layer to the given inputs.

Both inputs must have the same shape. Returns a tensor of shape [batch_size, seq_len, seq_len, out_features] given inputs of shape [batch_size, seq_len, in_features].

Trait Implementations

impl Debug for PairwiseBilinear

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.