Struct ISoftMaxLayer 

pub struct ISoftMaxLayer { /* private fields */ }

ISoftMaxLayer

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.

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)

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.

• 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

Get the axis along which softmax occurs.

See [setAxes()]

Trait Implementations

impl AsLayer for ISoftMaxLayer

fn as_layer(&self) -> &ILayer

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

impl AsLayerTyped for ISoftMaxLayer

const TYPE: LayerType = LayerType::kSOFTMAX

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.