Struct ReduceLayerParams 

pub struct ReduceLayerParams {
    pub mode: i32,
    pub epsilon: f32,
    pub axis: i32,
}

A layer that reduces the input using a specified operation.

.. code::

  y = ReduceLayer(x)

Requires 1 input and produces 1 output.

Input A blob that can, in general, have any rank. However, depending on the value of “axis” , there may be additional rank constraints. Output A blob with the same rank as the input, which has 1s on the dimensions specified in the parameter “axis”

 Values supported for axis are \[-1\], \[-2\], \[-3\], \[-2,-1\], \[-3,-2,-1\]
 and the equivalent positive values (depending on the rank of the input)
 For mode == 'ArgMax', axis must be \[-1\] or \[-2\] or \[-3\].

Fields

mode: i32

Specifies function used to reduce.

epsilon: f32

Used if mode is LOGSUM. Defaults to 1e-6 if not set or is set to 0.

axis: i32

The following mapping is used for interpreting this parameter: CHW = axis [-3, -2, -1], input must have rank at least 3. HW = axis [-2, -1], input must have rank at least 2. C = axis [-3] H = axis [-2] W = axis [-1]

Implementations

impl ReduceLayerParams

pub fn mode(&self) -> ReduceOperation

Returns the enum value of mode, or the default if the field is set to an invalid enum value.

pub fn set_mode(&mut self, value: ReduceOperation)

Sets mode to the provided enum value.

pub fn axis(&self) -> ReduceAxis

Returns the enum value of axis, or the default if the field is set to an invalid enum value.

pub fn set_axis(&mut self, value: ReduceAxis)

Sets axis to the provided enum value.

Trait Implementations

impl Clone for ReduceLayerParams

fn clone(&self) -> ReduceLayerParams

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ReduceLayerParams

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

Formats the value using the given formatter.

impl Default for ReduceLayerParams

fn default() -> Self

Returns the “default value” for a type.

impl Message for ReduceLayerParams

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 ReduceLayerParams

fn eq(&self, other: &ReduceLayerParams) -> 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 Copy for ReduceLayerParams

impl StructuralPartialEq for ReduceLayerParams