设置Tensor的形状。
1 2
template<typename T> void TensorShape::SetShape(std::vector<T> shape);
参数名
输入/输出
说明
shape
输入
Tensor的形状。
shape向量中各维度要求为正整数且单个或各项乘积需小于536,870,912（512 * 1024 * 1024），否则函数将抛出异常。