TensorShape
函数功能
TensorShape构造函数,创建TensorShape()对象。
函数原型
TensorShape::TensorShape() = default;
template<typename T> TensorShape::TensorShape (std::vector<T> shape);
参数说明
参数名 |
输入/输出 |
说明 |
---|---|---|
shape |
输入 |
Tensor的形状。 shape向量中各维度要求为正整数且单个或各项乘积需要小于52,428,8000(500MB),否则函数将抛出异常。 |
父主题: TensorShape