Enum drug::GlobalPool

pub enum GlobalPool {
    Average,
}

Type of pooling operation (currently there is only average pooling). TODO enum max pool, avg pool, sum pool, min pool Implements Operation. See Node constructor for full description.

Variants

Average

Reduces by taking the arithmetic mean

Trait Implementations

impl Debug for GlobalPool

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for GlobalPool

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

Deserialize this value from the given Serde deserializer.

impl Operation for GlobalPool

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 GlobalPool

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

Serialize this value into the given Serde serializer.