Struct ISoftMaxLayer 

pub struct ISoftMaxLayer { /* private fields */ }

! ! \class ISoftMaxLayer ! ! \brief A Softmax layer in a network definition. ! ! This layer applies a per-channel softmax to its input. ! ! The output size is the same as the input size. ! ! The following constraints must be satisfied to execute this layer on DLA: ! * Axis must be one of the channel or spatial dimensions. ! * There are two classes of supported input sizes: ! 1. Non-axis, non-batch dimensions are all 1 and the axis dimension is at most 8192. ! This is the recommended case for using softmax since it is the most accurate. ! 2. At least one non-axis, non-batch dimension greater than 1 and the axis dimension is at most 1024. ! Note that in this case, there may be some approximation error as the axis dimension size approaches ! the upper bound. See the TensorRT Developer Guide for more details on the approximation error. ! ! \warning Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI. !

Implementations

impl ISoftMaxLayer

pub fn setAxes(self: Pin<&mut ISoftMaxLayer>, axes: u32)

! ! \brief Set the axis along which softmax is computed. Currently, only one axis can be set. ! ! The axis is specified by setting the bit corresponding to the axis to 1. ! For example, consider an NCHW tensor as input. ! ! Bit 0 corresponds to the N dimension boolean. ! Bit 1 corresponds to the C dimension boolean. ! Bit 2 corresponds to the H dimension boolean. ! Bit 3 corresponds to the W dimension boolean. ! By default, softmax is performed on the axis which is the number of axes minus three. It is 0 if ! there are fewer than 3 axes. For example, if the input is NCHW, the default axis is C. If the input ! is NHW, then the default axis is N. ! ! For example, to perform softmax on axis R of a NPQRCHW input, set bit 3. ! ! \param axes The axis along which softmax is computed. ! Here axes is a bitmap. For example, when doing softmax along axis 0, bit 0 is set to 1, axes = 1 << axis ! = 1. !

pub fn getAxes(self: &ISoftMaxLayer) -> u32

! ! \brief Get the axis along which softmax occurs. ! ! \see setAxes() !

Trait Implementations

impl AsRef<ILayer> for ISoftMaxLayer

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

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

impl ExternType for ISoftMaxLayer

type Id = (n, v, i, n, f, e, r, _1, (), I, S, o, f, t, M, a, x, 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 ISoftMaxLayer

unsafe fn allocate_uninitialized_cpp_storage() -> *mut ISoftMaxLayer

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 ISoftMaxLayer)

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