TensorRT 8.5.3
nvinfer1::ISliceLayer Class Reference

Slices an input tensor into an output tensor based on the offset and strides. More...

#include <NvInfer.h>

Inheritance diagram for nvinfer1::ISliceLayer:
nvinfer1::ILayer nvinfer1::INoCopy

Public Member Functions

void setStart (Dims start) noexcept
 Set the start offset that the slice layer uses to create the output slice. More...
 
Dims getStart () const noexcept
 Get the start offset for the slice layer. More...
 
void setSize (Dims size) noexcept
 Set the dimensions of the output slice. More...
 
Dims getSize () const noexcept
 Get dimensions of the output slice. More...
 
void setStride (Dims stride) noexcept
 Set the stride for computing the output slice data. More...
 
Dims getStride () const noexcept
 Get the stride for the output slice. More...
 
void setMode (SliceMode mode) noexcept
 Set the slice mode. More...
 
SliceMode getMode () const noexcept
 Get the slice mode. More...
 
void setInput (int32_t index, ITensor &tensor) noexcept
 Append or replace an input of this layer with a specific tensor. 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...
 
ITensorgetInput (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...
 
ITensorgetOutput (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...
 

Protected Member Functions

virtual ~ISliceLayer () noexcept=default
 
- 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
 
INoCopyoperator= (INoCopy const &other)=delete
 
 INoCopy (INoCopy &&other)=delete
 
INoCopyoperator= (INoCopy &&other)=delete
 

Protected Attributes

apiv::VSliceLayer * mImpl
 
- Protected Attributes inherited from nvinfer1::ILayer
apiv::VLayer * mLayer
 

Detailed Description

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. Dynamic slice specifies one or more of start, size or stride as ITensors, by using ILayer::setInput to add a second, third, or fourth 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 input (third 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 Int32 shape tensors 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}}

When the sliceMode is kCLAMP or kREFLECT, for each input dimension, if its size is 0 then the corresponding output dimension must be 0 too.

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.
  • 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.
  • sliceMode is kDEFAULT.
  • Strides are 1 for all dimensions.
  • Slicing is not performed on the first dimension
  • The input tensor has four dimensions
Warning
Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI.

Constructor & Destructor Documentation

◆ ~ISliceLayer()

virtual nvinfer1::ISliceLayer::~ISliceLayer ( )
protectedvirtualdefaultnoexcept

Member Function Documentation

◆ getMode()

SliceMode nvinfer1::ISliceLayer::getMode ( ) const
inlinenoexcept

Get the slice mode.

See also
setMode()

◆ getSize()

Dims nvinfer1::ISliceLayer::getSize ( ) const
inlinenoexcept

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

◆ getStart()

Dims nvinfer1::ISliceLayer::getStart ( ) const
inlinenoexcept

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

◆ getStride()

Dims nvinfer1::ISliceLayer::getStride ( ) const
inlinenoexcept

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

◆ setInput()

void nvinfer1::ILayer::setInput ( int32_t  index,
ITensor tensor 
)
inlinenoexcept

Append or replace an input of this layer with a specific tensor.

Parameters
indexthe index of the input to modify.
tensorthe new input tensor

For a slice layer, the values 0-4 are valid. The indices are as follows:

  • 0: Tensor to be sliced.
  • 1: The start tensor to begin slicing, as a 1D Int32 shape tensor.
  • 2: The size tensor of the resulting slice, as a 1D Int32 shape tensor.
  • 3: The stride of the slicing operation, as a 1D Int32 shape tensor.
  • 4: Value for the kFILL slice mode. The fill value data type should have the same data phylum as input data type. And this input is disallowed for other modes.

Using the corresponding setter resets the input to null.

If this function is called with a value greater than 0, then the function getNbInputs() changes from returning 1 to index + 1.

◆ setMode()

void nvinfer1::ISliceLayer::setMode ( SliceMode  mode)
inlinenoexcept

Set the slice mode.

See also
getMode()

◆ setSize()

void nvinfer1::ISliceLayer::setSize ( Dims  size)
inlinenoexcept

Set the dimensions of the output slice.

Parameters
sizeThe 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

◆ setStart()

void nvinfer1::ISliceLayer::setStart ( Dims  start)
inlinenoexcept

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

Parameters
startThe 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

◆ setStride()

void nvinfer1::ISliceLayer::setStride ( Dims  stride)
inlinenoexcept

Set the stride for computing the output slice data.

Parameters
strideThe 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

Member Data Documentation

◆ mImpl

apiv::VSliceLayer* nvinfer1::ISliceLayer::mImpl
protected

The documentation for this class was generated from the following file: