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