Mojo function

managed_tensor_slice_to_ndbuffer

managed_tensor_slice_to_ndbuffer[spec: StaticTensorSpec[spec.dtype, spec.rank], //](tensor: ManagedTensorSlice[tensor.io_spec, static_spec=spec]) -> NDBuffer[spec.dtype, spec.rank, AnyOrigin[mut=(tensor.io_spec.input == IO.value).__bool__().__invert__()._mlir_value], spec.shape, spec.strides, address_space=spec.address_space]

Returns:

NDBuffer

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit