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
virtual void	setStart (Dims start)=0
	Set the start offset that the slice layer uses to create the output slice. More...

virtual Dims	getStart () const =0
	Get the start offset for the slice layer. More...

virtual void	setSize (Dims size)=0
	Set the dimensions of the output slice. More...

virtual Dims	getSize () const =0
	Get dimensions of the output slice. More...

virtual void	setStride (Dims stride)=0
	Set the stride for computing the output slice data. More...

virtual Dims	getStride () const =0
	Get the stride for the output slice. More...

void	setInput (int index, ITensor &tensor) _TENSORRT_OVERRIDE=0
	replace an input of this layer with a specific tensor. More...

Public Member Functions inherited from nvinfer1::ILayer
virtual LayerType	getType () const =0
	Return the type of a layer. More...

virtual void	setName (const char *name)=0
	Set the name of a layer. More...

virtual const char *	getName () const =0
	Return the name of a layer. More...

virtual int	getNbInputs () const =0
	Get the number of inputs of a layer.

virtual ITensor *	getInput (int index) const =0
	Get the layer input corresponding to the given index. More...

virtual int	getNbOutputs () const =0
	Get the number of outputs of a layer.

virtual ITensor *	getOutput (int index) const =0
	Get the layer output corresponding to the given index. More...

virtual void	setPrecision (DataType dataType)=0
	Set the computational precision of this layer. More...

virtual DataType	getPrecision () const =0
	get the computational precision of this layer More...

virtual bool	precisionIsSet () const =0
	whether the computational precision has been set for this layer More...

virtual void	resetPrecision ()=0
	reset the computational precision for this layer More...

virtual void	setOutputType (int index, DataType dataType)=0
	Set the output type of this layer. More...

virtual DataType	getOutputType (int index) const =0
	get the output type of this layer More...

virtual bool	outputTypeIsSet (int index) const =0
	whether the output type has been set for this layer More...

virtual void	resetOutputType (int index)=0
	reset the output type for this layer More...

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 create time via Dims and can use the get/set accessor functions of the ISliceLayer. Dynamic slice specifies the start and size dimensions at layer create time via ITensors and uses ILayer::setTensor to set the optional stride parameter after layer construction. An application can determine if the ISliceLayer is dynamic or static based on if there are 3 or 4 inputs(Dynamic) or 1 input(Static). When working on a shape tensor, a dynamic slace layer must have start, size, and stride specified at build time.

The slice layer selects for each dimension a start location from within the input tensor, and given the specified stride, copies strided elements to the output tensor. Start, Size, and Stride shape tensors must be DataType::kINT32.

For example using slice on a data tensor: input = {{0, 1}, {2, 3}, {4, 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

◆ getSize()

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

pure virtual

Get dimensions of the output slice.

Returns: The output dimension, or an invalid Dims structure.

If the SliceLayer is using dynamic inputs for the size parameter, this function returns an invalid Dims structure.

See also: setSize

◆ getStart()

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

pure virtual

Get the start offset for the slice layer.

Returns: The start offset, or an invalid Dims structure.

If the SliceLayer is using dynamic inputs for the start parameter, this function returns an invalid Dims structure.

See also: setStart

◆ getStride()

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

pure virtual

Get the stride for the output slice.

Returns: The slicing stride, or an invalid Dims structure.

If the SliceLayer is using dynamic inputs for the stride parameter, this function returns a invalid Dims structure.

See also: setStride

◆ setInput()

void nvinfer1::ISliceLayer::setInput	(	int	index,
		ITensor &	tensor
	)

pure virtual

replace an input of this layer with a specific tensor.

Parameters

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

Sets the input tensor for the given index. The index must be 0 for a static slice layer. A static slice layer is converted to a dynamic slice layer by calling setInput with an index > 0. A dynamic slice layer cannot be converted back to a static slice layer.

For a dynamic slice layer, the values 0-3 are valid. If an index > 0 is specified, all values between index 0 and that index must be dynamic tensors. The values larger than index can use static dimensions. For example, if an index of two is specified, the stride tensor can be set via setStride, but the start tensor must be specified via setInput as both size and start are converted to dynamic tensors. The indices in the dynamic case are as follows:

Index | Description 0 | Data or Shape tensor to be sliced. 1 | The start tensor to begin slicing, N-dimensional for Data, and 1-D for Shape. 2 | The size tensor of the resulting slice, N-dimensional for Data, and 1-D for Shape. 3 | The stride of the slicing operation, N-dimensional for Data, and 1-D for Shape.

If this function is called with a value greater than 0, then the function getNbInputs() changes from returning 1 to index + 1. When converting from static to dynamic slice layer, all unset tensors, between 1 and index + 1, are initialized to nullptr. It is an error to attempt to build a network that has any nullptr inputs.

Implements nvinfer1::ILayer.

◆ setSize()

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

pure virtual

Set the dimensions of the output slice.

Parameters

size	The dimensions of the output slice.

If the SliceLayer is using dynamic inputs for the size parameter, calling setSize() results in an error and does not update the dimensions.

See also: getSize

◆ setStart()

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

pure virtual

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 the SliceLayer is using dynamic inputs for the start parameter, calling setStart() results in an error and does not update the dimensions.

See also: getStart

◆ setStride()

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

pure virtual

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 the SliceLayer is using dynamic inputs for the stride parameter, calling setSlice() results in an error and does not update the dimensions.

See also: getStride

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

NvInfer.h

Public Member Functions

Detailed Description

Member Function Documentation

◆ getSize()

◆ getStart()

◆ getStride()

◆ setInput()

◆ setSize()

◆ setStart()

◆ setStride()