Tensor *AddInput(const std::string name, DataType type, Dims dim) noexcept;

函数功能

向网络添加输入张量。

函数原型

Tensor *AddInput(const std::string name, DataType type, Dims dim) noexcept;

约束说明

参数说明

参数名

输入/输出

说明

name

输入

张量的名称。

type

输入

张量中保存的数据类型。

dim

输入

张量的维度。

返回值说明

新的张量,如果添加失败时返回nullptr。