@Namespace(value="nvinfer1") @NoOffset @Properties(inherit=nvinfer.class) public class IGatherLayer extends ILayer
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
Constructor and Description |
---|
IGatherLayer(Pointer p)
Pointer cast constructor.
|
Modifier and Type | Method and Description |
---|---|
int |
getGatherAxis()
\brief Get the axis to gather on.
|
nvinfer.GatherMode |
getMode()
\brief Get the gather mode.
|
int |
getNbElementWiseDims()
\brief Get the number of leading dimensions of indices tensor to be handled elementwise.
|
void |
setGatherAxis(int axis)
\brief Set the axis used by GatherMode::kELEMENTS and GatherMode::kDEFAULT
The axis must be less than the number of dimensions in the data input.
|
void |
setMode(int mode) |
void |
setMode(nvinfer.GatherMode mode)
\brief Set the gather mode.
|
void |
setNbElementWiseDims(int elementWiseDims)
\brief Set the number of leading dimensions of indices tensor to be handled elementwise.
|
getInput, getMetadata, getName, getNbInputs, getNbOutputs, getOutput, getOutputType, getPrecision, getType, outputTypeIsSet, precisionIsSet, resetOutputType, resetPrecision, setInput, setMetadata, setMetadata, setName, setName, setOutputType, setOutputType, setPrecision, setPrecision
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, withDeallocator, zero
public IGatherLayer(Pointer p)
Pointer(Pointer)
.@NoException(value=true) public void setGatherAxis(int axis)
getGatherAxis()
@NoException(value=true) public int getGatherAxis()
setGatherAxis()
@NoException(value=true) public void setNbElementWiseDims(int elementWiseDims)
elementWiseDims
- number of dims to be handled as elementwise.
Default: 0
The value of nbElementWiseDims and GatherMode are checked during network validation:
GatherMode::kDEFAULT: nbElementWiseDims must be 0 if there is an implicit batch dimension. It can be 0 or 1 if
there is not an implicit batch dimension.
GatherMode::kND: nbElementWiseDims can be between 0 and one less than rank(data).
GatherMode::kELEMENT: nbElementWiseDims must be 0getNbElementWiseDims()
@NoException(value=true) public int getNbElementWiseDims()
setNbElementWiseDims()
@NoException(value=true) public void setMode(nvinfer.GatherMode mode)
getMode()
@NoException(value=true) public void setMode(@Cast(value="nvinfer1::GatherMode") int mode)
@NoException(value=true) public nvinfer.GatherMode getMode()
setMode()
Copyright © 2024. All rights reserved.