Flattens the input tensor into a 2D tensor based on the given axis.
- If the shape of self is (d_0, d_1,..., d_n), then the shape of the output out is (d_0*d_1...*d_(axis-1), d_axis*d_(axis+1)... *d_n).
- If axis is 0, the shape of the output out is (1, d_0*d_1... *d_n).
Each operator has calls. First, [object Object] is called to obtain the workspace size required for computation and the executor that contains the operator computation process. Then, [object Object] is called to perform computation.
[object Object]
[object Object]
Parameters:
[object Object]Returns:
[object Object]: status code. For details, see .The first-phase API performs input parameter validation. The following errors may be returned:
[object Object]
- Deterministic computation:
[object Object]defaults to a deterministic implementation.
The following example is for reference only. For details, see .
[object Object]