Tensor *GetInput(int32_t index) const noexcept;
函数功能
获取指定索引位置的输入张量。
函数原型
Tensor *GetInput(int32_t index) const noexcept;
参数说明
参数名 |
输入/输出 |
说明 |
---|---|---|
index |
输入 |
对应layer输入索引位置。 |
返回值说明
索引在有效范围,返回输入张量;索引不在有效范围,返回空指针。当前索引取值不支持负值。
父主题: class BaseLayer