昇腾社区首页
中文
注册

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

函数功能

向网络添加输入张量。

函数原型

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

约束说明

  • name必须为合法字符串,不能为空字符串,不能重复添加与之前相同的name,不能包含“\n”“\f”“\r”, "\v", "\t","\b","\u007f", " " 等转义字符,字符长度为:1~128。。
  • dim不可为空,且当前版本只支持添加静态的输入,即input的每一个维度都大于0。
  • dim维度必须在[1, 8]。
  • 当前不支持动态shape。

参数说明

参数名

输入/输出

说明

name

输入

张量的名称。

type

输入

张量中保存的数据类型。

dim

输入

张量的维度。

返回值说明

新的张量,如果添加失败则返回nullptr或抛出异常。