Mojo function
key
def key(idx: Scalar[DType.int], data: UnsafePointer[Float32, MutAnyOrigin], dim: Scalar[DType.int], split_dim: Scalar[DType.int]) -> Float32
Args:
- idx (
Scalar) - data (
UnsafePointer) - dim (
Scalar) - split_dim (
Scalar)
Returns:
Float32