nvinfer1::PluginTensorDesc 结构体参考

插件可能看到的输入或输出字段。 更多...

#include <NvInferRuntimePlugin.h>

公共属性
Dims	dims
	维度。 更多...
DataType	type
TensorFormat	format
	张量格式。 更多...
float	scale
	INT8 数据类型的缩放比例。 更多...

详细描述

插件可能看到的输入或输出字段。

仅当数据类型为 DataType::kINT8 时，缩放比例才有效。如果无效，TensorRT 将把该值设置为 -1.0F。

另请参阅

IPluginV2IOExt::supportsFormatCombination

IPluginV2IOExt::configurePlugin

成员数据文档

dims

Dims nvinfer1::PluginTensorDesc::dims

维度。

format

TensorFormat nvinfer1::PluginTensorDesc::format

张量格式。

scale

float nvinfer1::PluginTensorDesc::scale

INT8 数据类型的缩放比例。

type

DataType nvinfer1::PluginTensorDesc::type

警告

不支持 DataType::kBOOL 和 DataType::kUINT8。

---

此结构体的文档由以下文件生成

• NvInferRuntimePlugin.h