SetShape
Function Usage
Sets the tensor shape.
Prototype
template<typename T> void SetShape(std::vector<T> shape);
Parameter Description
Parameter |
Input/Output |
Description |
|---|---|---|
shape |
Input |
Tensor shape. Each dimension in the shape vector must be a positive integer. The maximum value of a single dimension and the product of all dimensions cannot exceed 536,870,912 (512 MB). Otherwise, the function will throw an exception. |
Parent topic: TensorShape