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

Unsqueeze

constructor(ctx: ExecutionContext, dim: Int)(source)

Parameters

T

The tensor data type

V

The value type

ctx

The execution context for tensor operations

dim

The dimension to add