Struct INonZeroLayer 

pub struct INonZeroLayer { /* private fields */ }

INonZero

A NonZero layer in a network.

This layer gets the positions of elements that are non-zero in the input. For boolean input, “non-zero” means “true”. Semantics are similar to ONNX NonZero.

The input may have type kFLOAT, kHALF, kINT32, or kBOOL.

The output is a matrix of type kINT32 or kINT64. For an input with dimensions [L1, L2, …, Lm], the output has dimensions [m,n], where n is the number of non-zero elements. I.e., each column denotes a m-D position.

The columns are lexically ordered. E.g., a column with [3,2,4,7] precedes a column with [3,2,5,6].

Tip: “compress” can be implemented with INonZero+IShuffle+Gather. For example, to compress a tensor x over axis k using mask vector v, use nonzero(v) to compute the subscripts, shuffle with reshape dimensions = [-1] to make the subscripts 1D, and then gather with the subscripts.

Implementations

impl INonZeroLayer

pub fn setIndicesType(self: Pin<&mut INonZeroLayer>, type_: DataType) -> bool

Set the indices type for the layer.

• type The DataType of the indices tensor.

true if set successfully, false otherwise.

Set the indices (the first output) type of the NonZero layer. Valid values are DataType::kINT32 and DataType::kINT64, otherwise an error occurs and the type is not updated.

pub fn getIndicesType(self: &INonZeroLayer) -> DataType

Return the NonZero layer indices type.

indices type set during layer creation or by setIndicesType(). The return value is the indices type of the NonZero layer. The default value is DataType::kINT32.

Trait Implementations

impl AsLayer for INonZeroLayer

fn as_layer(&self) -> &ILayer

fn as_layer_pin_mut(&mut self) -> Pin<&mut ILayer>

impl AsLayerTyped for INonZeroLayer

const TYPE: LayerType = LayerType::kNON_ZERO

impl AsRef<ILayer> for INonZeroLayer

fn as_ref(self: &INonZeroLayer) -> &ILayer

Converts this type into a shared reference of the (usually inferred) input type.

impl ExternType for INonZeroLayer

type Id = (n, v, i, n, f, e, r, _1, (), I, N, o, n, Z, e, r, o, L, a, y, e, r)

A type-level representation of the type’s C++ namespace and type name.

type Kind = Opaque

Either cxx::kind::Opaque or cxx::kind::Trivial.

impl MakeCppStorage for INonZeroLayer

unsafe fn allocate_uninitialized_cpp_storage() -> *mut INonZeroLayer

Allocates heap space for this type in C++ and return a pointer to that space, but do not initialize that space (i.e. do not yet call a constructor).

unsafe fn free_uninitialized_cpp_storage(arg0: *mut INonZeroLayer)

Frees a C++ allocation which has not yet had a constructor called.