Struct SliceStaticLayerParams 

pub struct SliceStaticLayerParams {
    pub begin_ids: Vec<i64>,
    pub begin_masks: Vec<bool>,
    pub end_ids: Vec<i64>,
    pub end_masks: Vec<bool>,
    pub strides: Vec<i64>,
    pub squeeze_masks: Vec<bool>,
}

A layer that extracts a slice of size (end - begin) / stride from the given input tensor. Support negative indexing and negative strides.

Requires 1 input and produces 1 output. Output rank is same as the input rank.

Value of beginIds, beginMasks, endIds, endMasks, strides are required parameters. Lengths of all the parameters must equal the rank of the input.

i-th element of “beginIds” is ignored and assumed to be 0 if the i-th element of “beginMasks” is True

i-th element of “endIds” is ignored and assumed to be -1 if the i-th element of “endMasks” is True

e.g.: if i-th element of “squeezeMasks” is set to True, only beginIds[i] would be sliced out, and all other masks and inputs are ignored.

e.g. (without squeezeMasks): input shape: (5, 5, 5) beginIds: [1, 2, 3] beginMasks: [True, False, True] endIds: [3, -3, 2] endMasks: [False, True, True] strides: [2, 2, 2] SqueezeMasks: [False, False, False] output shape: (2, 2, 3) This is equivalent to input[:3:2, 2::2, ::2]

e.g. (with squeezeMasks): input shape: (5, 5, 5) beginIds: [1, 2, 3] beginMasks: [True, False, True] endIds: [3, -3, 2] endMasks: [False, True, True] strides: [2, 2, 2] SqueezeMasks: [False, True, False] output shape: (2, 3) This is equivalent to input[:3:2, 2, ::2]

Fields

begin_ids: Vec<i64>

begin_masks: Vec<bool>

end_ids: Vec<i64>

end_masks: Vec<bool>

strides: Vec<i64>

squeeze_masks: Vec<bool>

Trait Implementations

impl Clone for SliceStaticLayerParams

fn clone(&self) -> SliceStaticLayerParams

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for SliceStaticLayerParams

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

Formats the value using the given formatter.

impl Default for SliceStaticLayerParams

fn default() -> Self

Returns the “default value” for a type.

impl Hash for SliceStaticLayerParams

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 SliceStaticLayerParams

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 SliceStaticLayerParams

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

impl StructuralPartialEq for SliceStaticLayerParams