Input(std::vector<int64_t> shape, DataType dtype, TensorFormat format = TensorFormat(TensorFormat::ND))
函数功能
静态输入构造函数。
函数原型
Input(std::vector<int64_t> shape, DataType dtype, TensorFormat format = TensorFormat(TensorFormat::ND))
约束说明
无
参数说明
参数名 |
输入/输出 |
说明 |
|---|---|---|
shape |
输入 |
静态输入的shape（在纯静态和动态分档模式中使用）；默认值：无。 |
dtype |
输入 |
输入的数据类型；默认值：DataType::UNKNOWN。 |
format |
输入 |
输入的数据排布；默认值：TensorFormat::ND。 |
返回值说明
Input对象。
父主题： Input