Skip to main content

/

Docs

Log in

Mojo struct

AnyMemory

A generic representation which can either be a Driver Tensor or Mojo object.

Implemented traits​

AnyType, CollectionElement, Copyable, Formattable, Movable

Methods​

__init__​

__init__(inout self: Self)

Default constructor for AnyMemory.

__init__(inout self: Self, owned device_tensor: DeviceTensor)

Creates AnyMemory from a DeviceTensor.

Args:

  • ​device_tensor (DeviceTensor): DeviceTensor to construct AnyMemory from.

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

Creates AnyMemory from a Tensor.

Args:

  • ​tensor (Tensor[type, rank]): Tensor to construct AnyMemory from.

__init__(inout self: Self, owned tensor: AnyTensor)

Creates AnyMemory from a AnyTensor.

Args:

  • ​tensor (AnyTensor): AnyTensor to construct AnyMemory from.

__init__(inout self: Self, owned value: AnyMojoValue)

Creates AnyMemory from AnyMojoValue.

Args:

  • ​value (AnyMojoValue): AnyMojoValue to construct AnyMemory from.

is_tensor​

is_tensor(self: Self) -> Bool

Check whether this contains a tensor.

Returns:

True if contains tensor.

take_tensor​

take_tensor(inout self: Self) -> AnyTensor

Take tensor from object. Further access to this object is undefined behavior.

Returns:

The tensor inside the memory as AnyTensor.

take​

take(inout self: Self) -> Self

The returned value takes self's resources and replaces them with default initialized values.

Returns:

Newly constructed AnyMemory that takes storage from this.

to_device_tensor​

to_device_tensor(owned self: Self) -> DeviceTensor

Consume this object and produces and instance of DeviceTensor. Only valid if this was created from DeviceTensor.

Returns:

DeviceTensor representation of AnyMemory.

to​

to[T: Movable](owned self: Self) -> $0

Consume this object and produces an instance of T. This doesn't do any type check beyond whether this is a AnyTensor or not, and if not assume this was created from T.

Returns:

An instance of type T.

take_value​

take_value(inout self: Self) -> AnyMojoValue

Take value from object. Further access to this object is undefined behavior.

Returns:

The value inside the memory as AnyMojoValue.

__str__​

__str__(self: Self) -> String

Gets this value as a string.

format_to​

format_to(self: Self, inout writer: Formatter)

Formats the string representation of this value to the provided formatter.

Args:

  • ​writer (Formatter): The formatter to write to.

Was this page helpful?