Struct FlattenLayerParams 

pub struct FlattenLayerParams {
    pub mode: i32,
}

A layer that flattens the input.

.. code::

  y = FlattenLayer(x)

Requires 1 input and produces 1 output.

Input A blob with rank greater than equal to 3. e.g.: Rank 4 blob represents [Batch, C, H, W] For ranks greater than 3, the leading dimensions, starting from 0 to -4 (inclusive), are all treated as batch. Output Same rank as the input, such that last two dimensions are both 1. e.g.: For rank 4 input, output shape is \[Batch, C * H * W, 1, 1\]

There are two X orders: CHANNEL_FIRST and CHANNEL_LAST. CHANNEL_FIRST does not require data to be rearranged, because row major ordering is used by internal storage. CHANNEL_LAST requires data to be rearranged.

Fields

mode: i32

Implementations

impl FlattenLayerParams

pub fn mode(&self) -> FlattenOrder

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: FlattenOrder)

Sets mode to the provided enum value.

Trait Implementations

impl Clone for FlattenLayerParams

fn clone(&self) -> FlattenLayerParams

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for FlattenLayerParams

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

Formats the value using the given formatter.

impl Default for FlattenLayerParams

fn default() -> Self

Returns the “default value” for a type.

impl Hash for FlattenLayerParams

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl Message for FlattenLayerParams

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 FlattenLayerParams

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

impl Eq for FlattenLayerParams

impl StructuralPartialEq for FlattenLayerParams