nvinfer1::IOneHotLayer Class Reference
A OneHot layer in a network definition. More...
#include <NvInfer.h>
Inheritance diagram for nvinfer1::IOneHotLayer:
Public Member Functions | |
| void | setAxis (int32_t axis) noexcept |
| Set the axis parameter. More... | |
| int32_t | getAxis () const noexcept |
| Get the value of the axis parameter. More... | |
 Public Member Functions inherited from nvinfer1::ILayer | |
| LayerType | getType () const noexcept |
| Return the type of a layer. More... | |
| void | setName (char const *name) noexcept |
| Set the name of a layer. More... | |
| char const * | getName () const noexcept |
| Return the name of a layer. More... | |
| int32_t | getNbInputs () const noexcept |
| Get the number of inputs of a layer. More... | |
| ITensor * | getInput (int32_t index) const noexcept |
| Get the layer input corresponding to the given index. More... | |
| int32_t | getNbOutputs () const noexcept |
| Get the number of outputs of a layer. More... | |
| ITensor * | getOutput (int32_t index) const noexcept |
| Get the layer output corresponding to the given index. More... | |
| void | setInput (int32_t index, ITensor &tensor) noexcept |
| Replace an input of this layer with a specific tensor. More... | |
| void | setPrecision (DataType dataType) noexcept |
| Set the computational precision of this layer. More... | |
| DataType | getPrecision () const noexcept |
| get the computational precision of this layer More... | |
| bool | precisionIsSet () const noexcept |
| whether the computational precision has been set for this layer More... | |
| void | resetPrecision () noexcept |
| reset the computational precision for this layer More... | |
| void | setOutputType (int32_t index, DataType dataType) noexcept |
| Set the output type of this layer. More... | |
| DataType | getOutputType (int32_t index) const noexcept |
| get the output type of this layer More... | |
| bool | outputTypeIsSet (int32_t index) const noexcept |
| whether the output type has been set for this layer More... | |
| void | resetOutputType (int32_t index) noexcept |
| reset the output type for this layer More... | |
| void | setMetadata (char const *metadata) noexcept |
| Set the metadata for this layer. More... | |
| char const * | getMetadata () const noexcept |
| Get the metadata of the layer. More... | |
Protected Attributes | |
| apiv::VOneHotLayer * | mImpl |
| Protected Attributes inherited from nvinfer1::ILayer | |
| apiv::VLayer * | mLayer |
Additional Inherited Members | |
| Protected Member Functions inherited from nvinfer1::ILayer | |
| virtual | ~ILayer () noexcept=default |
| Protected Member Functions inherited from nvinfer1::INoCopy | |
| INoCopy ()=default | |
| virtual | ~INoCopy ()=default |
| INoCopy (INoCopy const &other)=delete | |
| INoCopy & | operator= (INoCopy const &other)=delete |
| INoCopy (INoCopy &&other)=delete | |
| INoCopy & | operator= (INoCopy &&other)=delete |
Detailed Description
A OneHot layer in a network definition.
The OneHot layer has three input tensors: Indices, Values, and Depth, one output tensor: Output, and an axis attribute.
- Indices is an Int32 tensor that determines which locations in Output to set as on_value.
- Values is a two-element (rank=1) tensor that consists of [off_value, on_value]
- Depth is an Int32 shape tensor of rank 0, which contains the depth (number of classes) of the one-hot encoding. The depth tensor must be a build-time constant, and its value should be positive.
- Output is a tensor with rank = rank(indices)+1, where the added dimension contains the one-hot encoding. The data types of Output is equal to the Values data type.
- Axis is a scaler specifying to which dimension of the output one-hot encoding is added. Axis defaults to -1, that is the new dimension in the output is its final dimension. Valid range for axis is -rank(indices)-1 <= axis <= rank(indices).
The output is computed by copying off_values to all output elements, then setting on_value on the indices specified by the indices tensor. when axis = 0: output[indices[i, j, k], i, j, k] = on_value for all i, j, k and off_value otherwise.
when axis = -1: output[i, j, k, indices[i, j, k]] = on_value for all i, j, k and off_value otherwise.
- Warning
- Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI.
Member Function Documentation
◆ getAxis()
|
inlinenoexcept |
Get the value of the axis parameter.
◆ setAxis()
|
inlinenoexcept |
Set the axis parameter.
- See also
- IOneHotLayer
Member Data Documentation
◆ mImpl
|
protected |
The documentation for this class was generated from the following file:
