获取Tensor指定索引维度的大小。
1
size_t TensorShape::operator[] (uint32_t idx) const;
size_t &TensorShape::operator[] (uint32_t idx);
参数名
输入/输出
说明
idx
输入
Tensor维度索引值。
数据结构
size_t
Tensor指定维度的大小。