Method Flatten
Flatten(int)
Creates a TensorShape
by duplicating this
and reshaping to a 2D matrix. The dimensions before axis
are collapsed into the outer dimension and the remaining axes are collapsed into the inner dimension.
For example, if this
is (2, 3, 4), and the value of axis
is 2, the method returns (2 * 3, 4).
Declaration
public TensorShape Flatten(int axis = 1)
Parameters
Type | Name | Description |
---|---|---|
int | axis | The axis at which to flatten. |
Returns
Type | Description |
---|---|
TensorShape | The flattened tensor shape. |