基于偏移量和步幅将输入张量切片为输出张量。更多...

#include <NvInfer.h>

nvinfer1::ISliceLayer 的继承关系图

公共成员函数
void	setStart (Dims const &start) noexcept
	设置切片层用于创建输出切片的起始偏移量。更多...

Dims	getStart () const noexcept
	获取切片层的起始偏移量。更多...

void	setSize (Dims const &size) noexcept
	设置输出切片的维度。更多...

Dims	getSize () const noexcept
	获取输出切片的维度。更多...

void	setStride (Dims const &stride) noexcept
	设置用于计算输出切片数据的步幅。更多...

Dims	getStride () const noexcept
	获取输出切片的步幅。更多...

void	setMode (SampleMode mode) noexcept
	设置切片模式。更多...

SampleMode	getMode () const noexcept
	获取切片模式。更多...

void	setAxes (Dims const &axes) noexcept
	为此 ISliceLayer 设置轴。更多...

Dims	getAxes () const noexcept
	获取此 ISliceLayer 的轴。更多...

void	setInput (int32_t index, ITensor &tensor) noexcept
	为此层追加或替换特定张量的输入。更多...

从 nvinfer1::ILayer 继承的公共成员函数
LayerType	getType () const noexcept
	返回层的类型。更多...

void	setName (char const *name) noexcept
	设置层的名称。更多...

char const *	getName () const noexcept
	返回层的名称。更多...

int32_t	getNbInputs () const noexcept
	获取层的输入数量。更多...

ITensor *	getInput (int32_t index) const noexcept
	获取与给定索引对应的层输入。更多...

int32_t	getNbOutputs () const noexcept
	获取层的输出数量。更多...

ITensor *	getOutput (int32_t index) const noexcept
	获取与给定索引对应的层输出。更多...

void	setInput (int32_t index, ITensor &tensor) noexcept
	用特定张量替换此层的输入。更多...

void	setPrecision (DataType dataType) noexcept
	在弱类型网络中设置此层的首选或必需的计算精度。更多...

DataType	getPrecision () const noexcept
	获取此层的计算精度更多...

bool	precisionIsSet () const noexcept
	是否为此层设置了计算精度更多...

void	resetPrecision () noexcept
	重置此层的计算精度更多...

void	setOutputType (int32_t index, DataType dataType) noexcept
	在弱类型网络中设置此层的输出类型。更多...

DataType	getOutputType (int32_t index) const noexcept
	获取此层的输出类型更多...

bool	outputTypeIsSet (int32_t index) const noexcept
	是否为此层设置了输出类型更多...

void	resetOutputType (int32_t index) noexcept
	重置此层的输出类型更多...

void	setMetadata (char const *metadata) noexcept
	为此层设置元数据。更多...

char const *	getMetadata () const noexcept
	获取层的元数据。更多...

保护成员函数
virtual	~ISliceLayer () noexcept=default

从 nvinfer1::ILayer 继承的保护成员函数
virtual	~ILayer () noexcept=default

从 nvinfer1::INoCopy 继承的保护成员函数
	INoCopy ()=default

virtual	~INoCopy ()=default

	INoCopy (INoCopy const &other)=delete

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

	INoCopy (INoCopy &&other)=delete

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

保护属性
apiv::VSliceLayer *	mImpl

从 nvinfer1::ILayer 继承的保护属性
apiv::VLayer *	mLayer

详细描述

基于偏移量和步幅将输入张量切片为输出张量。

切片层有两种变体：静态和动态。静态切片在层创建时通过 Dims 指定起始、大小和步幅维度，并且可以使用 ISliceLayer 的 get/set 访问器函数。静态切片层还可以选择性地通过 ISliceLayer 的 get/set 访问器函数指定轴。动态切片通过使用 ILayer::setInput 添加第二个、第三个、第四个或第六个输入，从而将起始、大小、步幅或轴中的一个或多个指定为 ITensor。如果输入缺失或为空，则使用相应的 Dims。

应用程序可以根据大小或轴输入是否存在且非空来确定 ISliceLayer 是否具有动态输出形状。

切片层为每个维度从输入张量中选择一个起始位置，并使用指定的步幅跨输入张量将元素复制到输出张量。如果未通过 Dims 指定，则起始、大小和步幅张量必须是 Int32 或 Int64 类型的 1D 张量。

在张量上使用切片的示例：输入 = {{0, 2, 4}, {1, 3, 5}} 起始 = {1, 0} 大小 = {1, 2} 步幅 = {1, 2} 输出 = {{1, 5}}

如果提供了轴，则起始、结束和步幅必须与轴的长度相同，并指定要切片的维度子集。如果未提供轴，则起始、结束和步幅的长度必须与输入张量的秩相同。

在使用轴指定的张量上使用切片的示例：输入 = {{0, 2, 4}, {1, 3, 5}} 起始 = {1} 大小 = {2} 步幅 = {1} 轴 = {1} 输出 = {{2, 4}, {3, 5}}

当 sampleMode 为 kCLAMP 或 kREFLECT 时，对于每个输入维度，如果其大小为 0，则相应的输出维度也必须为 0。

当 sampleMode 为 kFILL 时，切片层的第五个输入用于确定要填充超出边界索引的值。在任何其他 sampleMode 中指定第五个输入都是错误的。

如果满足以下条件，则切片层可以生成形状张量

起始、大小和步幅是构建时常量，可以是静态 Dims 或常量输入张量。
轴（如果提供）是构建时常量，可以是静态 Dims 或常量输入张量。
输出张量中的元素数量不超过 2 * Dims::MAX_DIMS。

如果输出是形状张量，则输入张量是形状张量。

必须满足以下约束才能在 DLA 上执行此层

起始、大小和步幅是构建时常量，可以是静态 Dims 或常量输入张量。
轴（如果提供）是构建时常量，可以是静态 Dims 或常量输入张量。
sampleMode 为 kDEFAULT、kWrap 或 kFILL。
所有维度的步幅均为 1。
切片不在第一个维度上执行。
输入张量具有四个维度。
对于 kFILL sliceMode，填充值输入是 IConstantLayer 的标量输出，值为 0，且未被任何其他层使用。

警告: 请勿从此类继承，因为这样做会破坏 API 和 ABI 的向前兼容性。

构造函数 & 析构函数文档

◆ ~ISliceLayer()

virtual nvinfer1::ISliceLayer::~ISliceLayer ( )

protectedvirtualdefaultnoexcept

成员函数文档

◆ getAxes()

Dims nvinfer1::ISliceLayer::getAxes ( ) const

inlinenoexcept

获取此 ISliceLayer 的轴。

返回: 此切片的起始、结束和步幅参数应用到的轴。

如果第六个输入存在且非空，则此函数返回一个 nbDims = -1 的 Dims。

参见: setAxes

◆ getMode()

SampleMode nvinfer1::ISliceLayer::getMode ( ) const

inlinenoexcept

获取切片模式。

参见: setMode()

◆ getSize()

Dims nvinfer1::ISliceLayer::getSize ( ) const

inlinenoexcept

获取输出切片的维度。

返回: 输出维度，或无效的 Dims 结构。

如果第三个输入存在且非空，则此函数返回一个 nbDims = -1 的 Dims。

参见: setSize

◆ getStart()

Dims nvinfer1::ISliceLayer::getStart ( ) const

inlinenoexcept

获取切片层的起始偏移量。

返回: 起始偏移量，或无效的 Dims 结构。

如果第二个输入存在且非空，则此函数返回一个 nbDims = -1 的 Dims。

参见: setStart

◆ getStride()

Dims nvinfer1::ISliceLayer::getStride ( ) const

inlinenoexcept

获取输出切片的步幅。

返回: 切片步幅，或无效的 Dims 结构。

如果第四个输入存在且非空，则此函数返回一个 nbDims = -1 的 Dims。

参见: setStride

◆ setAxes()

void nvinfer1::ISliceLayer::setAxes ( Dims const & axes )

inlinenoexcept

为此 ISliceLayer 设置轴。

参数

axes	此切片的起始、结束和步幅参数应用到的轴。

如果第六个输入已用于创建此层，则此方法会将该输入重置为空。

参见: getAxes

◆ setInput()

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

inlinenoexcept

为此层追加或替换特定张量的输入。

参数

index	要修改的输入的索引。
tensor	新的输入张量

对于切片层，值 0-4 有效。索引如下：

0：要切片的张量。
1：开始切片的起始张量，作为 Int32 或 Int64 类型的 1D 张量。
2：结果切片的大小张量，作为 Int32 或 Int64 类型的 1D 张量。
3：切片操作的步幅，作为 Int32 或 Int64 类型的 1D 张量。
4：kFILL 切片模式的 Value。填充值数据类型应与输入数据类型相同，或者可以隐式转换为输入数据类型。kFLOAT、kHALF、kINT8 和 kFP8 数据类型之间支持隐式数据类型转换。此输入在其他模式下不允许使用。
5：轴张量，指示起始、大小和步幅应应用于的相应轴，作为 1D 张量或 Int32 或 Int64 类型。轴的负值表示从输入张量的背面开始索引。值必须唯一，并且在 [-rank(input), rank(input)-1] 的区间内。

使用相应的 setter 会将输入重置为空。

如果使用大于 0 的值调用此函数，则函数 getNbInputs() 从返回 1 更改为 index + 1。

◆ setMode()

void nvinfer1::ISliceLayer::setMode ( SampleMode mode )

inlinenoexcept

设置切片模式。

参见: getMode()

◆ setSize()

void nvinfer1::ISliceLayer::setSize ( Dims const & size )

inlinenoexcept

设置输出切片的维度。

参数

size 输出切片的维度。

如果第三个输入已用于创建此层，则此方法会将该输入重置为空。

参见: getSize

◆ setStart()

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

inlinenoexcept

设置切片层用于创建输出切片的起始偏移量。

参数

start 从输入张量读取数据的起始偏移量。

如果第二个输入已用于创建此层，则此方法会将该输入重置为空。

参见: getStart

◆ setStride()

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

inlinenoexcept

设置用于计算输出切片数据的步幅。

参数

stride 用于计算要存储在输出切片中的值的步幅维度。

如果第四个输入已用于创建此层，则此方法会将该输入重置为空。

参见: getStride

成员数据文档

◆ mImpl

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

protected

此类文档从以下文件生成：

NvInfer.h

公共成员函数

保护成员函数

保护属性

详细描述

构造函数 & 析构函数文档

◆ ~ISliceLayer()

成员函数文档

◆ getAxes()

◆ getMode()

◆ getSize()

◆ getStart()

◆ getStride()

◆ setAxes()

◆ setInput()

◆ setMode()

◆ setSize()

◆ setStart()

◆ setStride()

成员数据文档

◆ mImpl