TensorRT  8.0.2
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 (const char *name) noexcept
 Set the name of a layer. More...
 
const char * getName () const noexcept
 Return the name of a layer. More...
 
int32_t getNbInputs () const noexcept
 Get the number of inputs of a layer.
 
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.
 
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 Attributes

apiv::VSliceLayermImpl
 
- Protected Attributes inherited from nvinfer1::ILayer
apiv::VLayermLayer
 

Additional Inherited Members

- Protected Member Functions inherited from nvinfer1::INoCopy
 INoCopy (const INoCopy &other)=delete
 
INoCopyoperator= (const INoCopy &other)=delete
 
 INoCopy (INoCopy &&other)=delete
 
INoCopyoperator= (INoCopy &&other)=delete
 

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::setTensor 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.

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 computable by constant folding.
  • The number of elements in the output tensor does not exceed 2*Dims::MAX_DIMS.

For example using slice on a tensor: input = {{0, 2, 4}, {1, 3, 5}} start = {1, 0} size = {1, 2} stride = {1, 2} output = {{1, 5}}

Warning
Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI.

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
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-3 are valid. The values 1-3 override start, size or stride dimensions, respectively. Conversely, this input tensor can be overridden via appropriate set call. The indices are as follows:

  • 0: Data or Shape 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.

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

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