Skip to main content

Mojo struct

EngineTensorView

A non-owning register_passable view of a tensor that does runtime type checking.

CAUTION: Make sure the source tensor outlives the view.

Implemented traits

AnyType, Copyable, Movable

Methods

__init__

__init__[type: DType](inout self: Self, tensor: Tensor[type])

Creates a non-owning view of given Tensor.

Parameters:

  • type (DType): DType of the tensor.

Args:

  • tensor (Tensor[type]): Tensor backing the view.

data

data[type: DType](self: Self) -> UnsafePointer[SIMD[$0, 1], 0, 0, alignof[::AnyType,__mlir_type.!kgen.target]() if triple_is_nvidia_cuda() else 1]

Returns pointer to the start of tensor.

Parameters:

  • type (DType): Expected type of tensor.

Returns:

UnsafePointer of given type.

Raises:

If the given type does not match the type of tensor.

unsafe_ptr

unsafe_ptr(self: Self) -> UnsafePointer[NoneType, 0, 0, alignof[::AnyType,__mlir_type.!kgen.target]() if triple_is_nvidia_cuda() else 1]

Returns type erased pointer to the start of tensor.

Returns:

UnsafePointer of invalid type.

spec

spec(self: Self) -> TensorSpec

Returns the spec of tensor backing the view.

Returns:

Stdlib TensorSpec of the tensor.