skainet-data-transform/sk.ainet.data.transform/Flatten/Flatten

Flatten

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

Parameters

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)

Type Parameters

T

The tensor data type

V

The value type