下载
中文
注册

TensorShape

函数功能

TensorShape构造函数,创建TensorShape()对象。

函数原型

TensorShape::TensorShape() = default;
template<typename T>
TensorShape::TensorShape (std::vector<T> shape);

参数说明

参数名

输入/输出

说明

shape

输入

Tensor的形状。

shape向量中各维度要求为正整数且单个或各项乘积需要小于52,428,8000(500MB),否则函数将抛出异常。‬