@Namespace(value="nvinfer1") @NoOffset @Properties(inherit=nvinfer.class) public class ISoftMaxLayer extends ILayer
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
Constructor and Description |
---|
ISoftMaxLayer(Pointer p)
Pointer cast constructor.
|
Modifier and Type | Method and Description |
---|---|
int |
getAxes()
\brief Get the axis along which softmax occurs.
|
void |
setAxes(int axes)
\brief Set the axis along which softmax is computed.
|
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 ISoftMaxLayer(Pointer p)
Pointer(Pointer)
.@NoException(value=true) public void setAxes(@Cast(value="uint32_t") int axes)
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.@Cast(value="uint32_t") @NoException(value=true) public int getAxes()
setAxes()
Copyright © 2024. All rights reserved.