【TensorRT】TensorRT学习笔记之ITensor类

TensorRT中的ITensor类是一个非常重要的类，它代表了神经网络中的张量。以下是对TensorRT ITensor类的详细说明：

一、概述

ITensor类是TensorRT库中用于表示张量的核心类。在神经网络中，张量是一个多维数组，用于存储输入数据、网络参数、中间结果和输出数据等。ITensor类提供了丰富的成员函数和变量来操作和管理张量，如设置和获取张量的名称、维度、数据类型等。

二、成员函数

ITensor类提供了以下主要成员函数：

1. setName(const char name)：设置张量的名称。这有助于在调试和可视化过程中识别张量。
2. getName() const：获取张量的名称。
3. setDimensions(Dims dimensions)：设置张量的维度。Dims是一个结构体，包含了张量的维度信息。
4. getDimensions() const：获取张量的维度。
5. setType(DataType type)：设置张量的数据类型。TensorRT支持多种数据类型，如FP32、FP16、INT8等。
6. getType() const：获取张量的数据类型。