Struct drug::nodes::Embedding

pub struct Embedding();

Trainable embedding operation, given an index and a 2d-array of embedding vectors, index into the embedding vectors. FIXME drug hardcodes ArrayD<f32> inside the graph so the index should be a batch_size length arrayD<f32> where the values are integers.

Trait Implementations

impl Debug for Embedding

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for Embedding

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
    __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Operation for Embedding

fn eval(&self, inputs: &[ArrayViewD<'_, f32>]) -> ArrayD<f32>

Mutates Outputs based on inputs. TODO consider modifying output ArrayD in place

fn grad(
    &self,
    inputs: &[ArrayViewD<'_, f32>],
    loss: ArrayViewD<'_, f32>
) -> Vec<ArrayD<f32>>

Returns gradients of inputs wrt outputs. Note the inputs and output vectors should be the same length. TODO consider modifying output ArrayDs in place

impl Serialize for Embedding

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>where
    __S: Serializer,

Serialize this value into the given Serde serializer.