org.bytedeco.tensorrt.nvinfer

Class ISliceLayer

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.tensorrt.nvinfer.INoCopy

org.bytedeco.tensorrt.nvinfer.ILayer

org.bytedeco.tensorrt.nvinfer.ISliceLayer

All Implemented Interfaces:

AutoCloseable

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

\brief Slices an input tensor into an output tensor based on the offset and strides. The slice layer has two variants, static and dynamic. Static slice specifies the start, size, and stride dimensions at layer creation time via Dims and can use the get/set accessor functions of the ISliceLayer. Static slice layers can also optionally specify axes through the get/set accessor functions of the ISliceLayer. Dynamic slice specifies one or more of start, size, stride, or axes as ITensors, by using ILayer::setInput to add a second, third, fourth, or sixth input respectively. The corresponding Dims are used if an input is missing or null. An application can determine if the ISliceLayer has a dynamic output shape based on whether the size or axes input is present and non-null. The slice layer selects for each dimension a start location from within the input tensor, and copies elements to the output tensor using the specified stride across the input tensor. Start, size, and stride tensors must be 1D tensors of type Int32 or Int64 if not specified via Dims. An example of using slice on a tensor: input = {{0, 2, 4}, {1, 3, 5}} start = {1, 0} size = {1, 2} stride = {1, 2} output = {{1, 5}} If axes are provided then starts, ends, and strides must have the same length as axes and specifies a subset of dimensions to slice. If axes are not provided, starts, ends, and strides must be of the same length as the rank of the input tensor. An example of using slice on a tensor with axes specified: input = {{0, 2, 4}, {1, 3, 5}} start = {1} size = {2} stride = {1} axes = {1} output = {{2, 4}, {3, 5}} When the sampleMode is kCLAMP or kREFLECT, for each input dimension, if its size is 0 then the corresponding output dimension must be 0 too. When the sampleMode is kFILL, the fifth input to the slice layer is used to determine the value to fill in out-of-bound indices. It is an error to specify the fifth input in any other sampleMode. A slice layer can produce a shape tensor if the following conditions are met: * start, size, and stride are build time constants, either as static Dims or as constant input tensors. * axes, if provided, are build time constants, either as static Dims or as a constant input tensor. * The number of elements in the output tensor does not exceed 2 * Dims::MAX_DIMS. The input tensor is a shape tensor if the output is a shape tensor. The following constraints must be satisfied to execute this layer on DLA: * start, size, and stride are build time constants, either as static Dims or as constant input tensors. * axes, if provided, are build time constants, either as static Dims or as a constant input tensor. * sampleMode is kDEFAULT, kWRAP, or kFILL. * Strides are 1 for all dimensions. * Slicing is not performed on the first dimension. * The input tensor has four dimensions. * For kFILL sliceMode, the fill value input is a scalar output of an IConstantLayer with value 0 that is not consumed by any other layer. \warning Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI.

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
ISliceLayer(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
Dims64	getAxes() \brief Get the axes for this ISliceLayer.
nvinfer.SampleMode	getMode() \brief Get the slice mode.
Dims64	getSize() \brief Get dimensions of the output slice.
Dims64	getStart() \brief Get the start offset for the slice layer.
Dims64	getStride() \brief Get the stride for the output slice.
void	setAxes(Dims64 axes) \brief Set the axes for this ISliceLayer.
void	setMode(int mode)
void	setMode(nvinfer.SampleMode mode) \brief Set the slice mode.
void	setSize(Dims64 size) \brief Set the dimensions of the output slice.
void	setStart(Dims64 start) \brief Set the start offset that the slice layer uses to create the output slice.
void	setStride(Dims64 stride) \brief Set the stride for computing the output slice data.

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, withDeallocator, zero

Methods inherited from class java.lang.Object

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

Constructor Detail

ISliceLayer

public ISliceLayer(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

Method Detail

setStart

@NoException(value=true)
public void setStart(@Cast(value="const nvinfer1::Dims*") @ByRef
                                               Dims64 start)

\brief Set the start offset that the slice layer uses to create the output slice.

Parameters:

start - The start offset to read data from the input tensor. If a second input had been used to create this layer, that input is reset to null by this method.

See Also:

getStart

getStart

@ByVal
 @Cast(value="nvinfer1::Dims*")
 @NoException(value=true)
public Dims64 getStart()

\brief Get the start offset for the slice layer.

Returns:

The start offset, or an invalid Dims structure. If the second input is present and non-null, this function returns a Dims with nbDims = -1.

See Also:

setStart

setSize

@NoException(value=true)
public void setSize(@Cast(value="const nvinfer1::Dims*") @ByRef
                                              Dims64 size)

\brief Set the dimensions of the output slice.

Parameters:

size - The dimensions of the output slice. If a third input had been used to create this layer, that input is reset to null by this method.

See Also:

getSize

getSize

@ByVal
 @Cast(value="nvinfer1::Dims*")
 @NoException(value=true)
public Dims64 getSize()

\brief Get dimensions of the output slice.

Returns:

The output dimension, or an invalid Dims structure. If the third input is present and non-null, this function returns a Dims with nbDims = -1.

See Also:

setSize

setStride

@NoException(value=true)
public void setStride(@Cast(value="const nvinfer1::Dims*") @ByRef
                                                Dims64 stride)

\brief Set the stride for computing the output slice data.

Parameters:

stride - The dimensions of the stride to compute the values to store in the output slice. If a fourth input had been used to create this layer, that input is reset to null by this method.

See Also:

getStride

getStride

@ByVal
 @Cast(value="nvinfer1::Dims*")
 @NoException(value=true)
public Dims64 getStride()

\brief Get the stride for the output slice.

Returns:

The slicing stride, or an invalid Dims structure. If the fourth input is present and non-null, this function returns a Dims with nbDims = -1.

See Also:

setStride

setMode

@NoException(value=true)
public void setMode(nvinfer.SampleMode mode)

\brief Set the slice mode.

See Also:

getMode()

setMode

@NoException(value=true)
public void setMode(@Cast(value="nvinfer1::SampleMode")
                                              int mode)

getMode

@NoException(value=true)
public nvinfer.SampleMode getMode()

\brief Get the slice mode.

See Also:

setMode()

setAxes

@NoException(value=true)
public void setAxes(@Cast(value="const nvinfer1::Dims*") @ByRef
                                              Dims64 axes)

\brief Set the axes for this ISliceLayer.

Parameters:

axes - The axes on which the starts, ends, and strides parameters of the slice apply to. If a sixth input had been used to create this layer, that input is reset to null by this method.

See Also:

getAxes

getAxes

@ByVal
 @Cast(value="nvinfer1::Dims*")
 @NoException(value=true)
public Dims64 getAxes()

\brief Get the axes for this ISliceLayer.

Returns:

The axes on which the starts, ends, and strides parameters of this slice apply to. If the sixth input is present and non-null, this function returns a Dims with nbDims = -1.

See Also:

setAxes