Flatten

constructor(ctx: ExecutionContext, startDim: Int = 0, endDim: Int = -1)(source)

Parameters

T

The tensor data type

V

The value type

ctx

The execution context for tensor operations

startDim

The first dimension to flatten (default: 0)

endDim

The last dimension to flatten (default: -1, meaning last dimension)