Struct UnaryFunctionLayerParams 

pub struct UnaryFunctionLayerParams {
    pub type: i32,
    pub alpha: f32,
    pub epsilon: f32,
    pub shift: f32,
    pub scale: f32,
}

A layer that applies a unary function.

.. code::

  y = UnaryFunctionLayer(x)

Requires 1 input and produces 1 output.

Input A blob with no rank constraints. Output A blob with the same shape as the input.

The input is first modified by shifting and scaling:

.. math:: x \leftarrow \text{scale} \cdot x + \text{shift}

Fields

type: i32

The type of unary function.

alpha: f32

A constant used in POWER and THRESHOLD functions.

epsilon: f32

A small constant to avoid division by 0 while normalizing variance. Defaults to 1e-6 if not set or set to 0.

shift: f32

Input is shifted by this amount before the unary function is applied. Defaults to 0.0 if not set.

scale: f32

Input is scaled by this amount before the unary function is applied. Defaults to 1.0 if not set or set to 0.

Implementations

impl UnaryFunctionLayerParams

pub fn type(&self) -> Operation

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

pub fn set_type(&mut self, value: Operation)

Sets type to the provided enum value.

Trait Implementations

impl Clone for UnaryFunctionLayerParams

fn clone(&self) -> UnaryFunctionLayerParams

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for UnaryFunctionLayerParams

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

Formats the value using the given formatter.

impl Default for UnaryFunctionLayerParams

fn default() -> Self

Returns the “default value” for a type.

impl Message for UnaryFunctionLayerParams

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 UnaryFunctionLayerParams

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

impl StructuralPartialEq for UnaryFunctionLayerParams