org.bytedeco.tensorrt.nvinfer

Class INonZeroLayer

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.tensorrt.nvinfer.INoCopy

org.bytedeco.tensorrt.nvinfer.ILayer

org.bytedeco.tensorrt.nvinfer.INonZeroLayer

All Implemented Interfaces:

AutoCloseable

@Namespace(value="nvinfer1")
 @NoOffset
 @Properties(inherit=nvinfer.class)
public class INonZeroLayer
extends ILayer

\class INonZero \brief 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.

Nested Class Summary

Nested classes/interfaces inherited from class org.bytedeco.javacpp.Pointer

Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter

Field Summary

Fields inherited from class org.bytedeco.javacpp.Pointer

address, capacity, limit, position

Constructor Summary

Constructors

Constructor and Description
INonZeroLayer(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
nvinfer.DataType	getIndicesType() \brief Return the NonZero layer indices type.
boolean	setIndicesType(int type)
boolean	setIndicesType(nvinfer.DataType type) \brief Set the indices type for the layer.

Methods inherited from class org.bytedeco.tensorrt.nvinfer.ILayer

getInput, getMetadata, getName, getNbInputs, getNbOutputs, getOutput, getOutputType, getPrecision, getType, outputTypeIsSet, precisionIsSet, resetOutputType, resetPrecision, setInput, setMetadata, setMetadata, setName, setName, setOutputType, setOutputType, setPrecision, setPrecision

Methods inherited from class org.bytedeco.javacpp.Pointer

address, asBuffer, asByteBuffer, availablePhysicalBytes, calloc, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, formatBytes, free, getDirectBufferAddress, getPointer, getPointer, getPointer, getPointer, hashCode, interruptDeallocatorThread, isNull, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetAddress, offsetof, offsetof, parseBytes, physicalBytes, physicalBytesInaccurate, position, position, put, realloc, referenceCount, releaseReference, retainReference, setNull, sizeof, sizeof, toString, totalBytes, totalCount, totalPhysicalBytes, trimMemory, withDeallocator, zero

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Detail

INonZeroLayer

public INonZeroLayer(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

Method Detail

setIndicesType

@Cast(value="bool")
 @NoException(value=true)
public boolean setIndicesType(nvinfer.DataType type)

\brief Set the indices type for the layer.

Parameters:

type - The DataType of the indices tensor.

Returns:

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.

setIndicesType

@Cast(value="bool")
 @NoException(value=true)
public boolean setIndicesType(@Cast(value="nvinfer1::DataType")
                                                                            int type)

getIndicesType

@NoException(value=true)
public nvinfer.DataType getIndicesType()

\brief Return the NonZero layer indices type.

Returns:

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.