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

#include <NvInfer.h>

Inheritance diagram for nvinfer1::ISliceLayer:

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.

ITensor *	getInput (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.

ITensor *	getOutput (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::VSliceLayer *	mImpl

Protected Attributes inherited from nvinfer1::ILayer
apiv::VLayer *	mLayer

Additional Inherited Members
Protected Member Functions inherited from nvinfer1::INoCopy
	INoCopy (const INoCopy &other)=delete

INoCopy &	operator= (const INoCopy &other)=delete

	INoCopy (INoCopy &&other)=delete

INoCopy &	operator= (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}}

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.

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	(	int32_t	index,
		ITensor &	tensor
	)

inlinenoexcept

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

Parameters

index	the index of the input to modify.
tensor	the new input tensor

For a slice layer, the values 0-4 are valid. 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.
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

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

◆ setStart()

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

inlinenoexcept

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

◆ setStride()

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

inlinenoexcept

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

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

NvInfer.h

Public Member Functions

Protected Attributes

Additional Inherited Members

Detailed Description

Member Function Documentation

◆ getMode()

◆ getSize()

◆ getStart()

◆ getStride()

◆ setInput()

◆ setMode()

◆ setSize()

◆ setStart()

◆ setStride()