Struct EmbeddingLayerParams 

pub struct EmbeddingLayerParams {
    pub input_dim: u64,
    pub output_channels: u64,
    pub has_bias: bool,
    pub weights: Option<WeightParams>,
    pub bias: Option<WeightParams>,
}

A layer that performs a matrix lookup and optionally adds a bias. The weights matrix is stored with dimensions [outputChannels, inputDim].

.. code::

  y = EmbeddingLayer(x)

Requires 1 input and produces 1 output.

Input Input values must be in the range \[0, inputDim - 1\].

 Input must have rank equal to 4 or 5, such that the last 3 dimensions are all 1.
 rank 4: shape (x1, 1, 1, 1). x1 is effectively the batch/sequence length.
 rank 5: shape (x1, x2 , 1, 1, 1). x1 * x2 is effectively the combined batch/sequence length.

Output Output rank is same as the input rank. Please see input description above. rank 4: shape (x1, outputChannels, 1, 1) rank 5: shape (x1, x2, outputChannels, 1, 1)

Fields

input_dim: u64

Size of the input dictionary.

output_channels: u64

Size of the output vectors.

has_bias: bool

Whether a bias is added or not.

weights: Option<WeightParams>

2-D weights of dimensions [outputChannels, inputDim].

bias: Option<WeightParams>

Bias of size [outputChannels].

Trait Implementations

impl Clone for EmbeddingLayerParams

fn clone(&self) -> EmbeddingLayerParams

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for EmbeddingLayerParams

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

Formats the value using the given formatter.

impl Default for EmbeddingLayerParams

fn default() -> Self

Returns the “default value” for a type.

impl Message for EmbeddingLayerParams

fn encoded_len(&self) -> usize

Returns the encoded length of the message without a length delimiter.

fn clear(&mut self)

Clears the message, resetting all fields to their default.

fn encode(&self, buf: &mut impl BufMut) -> Result<(), EncodeError>

where Self: Sized,

Encodes the message to a buffer.

fn encode_to_vec(&self) -> Vec<u8>

where Self: Sized,

Encodes the message to a newly allocated buffer.

fn encode_length_delimited( &self, buf: &mut impl BufMut, ) -> Result<(), EncodeError>

where Self: Sized,

Encodes the message with a length-delimiter to a buffer.

fn encode_length_delimited_to_vec(&self) -> Vec<u8>

where Self: Sized,

Encodes the message with a length-delimiter to a newly allocated buffer.

fn decode(buf: impl Buf) -> Result<Self, DecodeError>

where Self: Default,

Decodes an instance of the message from a buffer.

fn decode_length_delimited(buf: impl Buf) -> Result<Self, DecodeError>

where Self: Default,

Decodes a length-delimited instance of the message from the buffer.

fn merge(&mut self, buf: impl Buf) -> Result<(), DecodeError>

where Self: Sized,

Decodes an instance of the message from a buffer, and merges it into self.

fn merge_length_delimited(&mut self, buf: impl Buf) -> Result<(), DecodeError>

where Self: Sized,

Decodes a length-delimited instance of the message from buffer, and merges it into self.

impl PartialEq for EmbeddingLayerParams

fn eq(&self, other: &EmbeddingLayerParams) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl StructuralPartialEq for EmbeddingLayerParams