Struct 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.