operator[]
函数功能
获取Tensor指定索引维度的大小。
函数原型
size_t TensorShape::operator[] (uint32_t idx) const;
size_t &TensorShape::operator[] (uint32_t idx);
参数说明
参数名 |
输入/输出 |
说明 |
|---|---|---|
idx |
输入 |
Tensor维度索引值。 |
返回参数说明
数据结构 |
说明 |
|---|---|
size_t |
Tensor指定维度的大小。 |
父主题： TensorShape