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. ‬