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或抛出异常。
父主题: class Network