Table of Contents

Mojo struct

Matrix

@memory_only
struct Matrix[Type: DType]

Parameters

  • Type (DType)

Fields

  • data (Pointer[Scalar[Type], MutUntrackedOrigin])
  • layout (MatLayout)

Implemented traits

AnyType, Deinitable, Movable

Methods

__init__

fn def __init__(out self, shape: Tuple[Int, Int])

Args:

  • shape (Tuple[Int, Int])
  • self (Self)

Returns:

Self

fn def __init__(out self, data: Pointer[Scalar[Type], MutUntrackedOrigin], var layout: MatLayout)

Args:

  • data (Pointer[Scalar[Type], MutUntrackedOrigin])
  • layout (MatLayout)
  • self (Self)

Returns:

Self

fn def __init__(out self, data: Pointer[Scalar[Type], MutUntrackedOrigin], shape: Tuple[Int, Int])

Args:

  • data (Pointer[Scalar[Type], MutUntrackedOrigin])
  • shape (Tuple[Int, Int])
  • self (Self)

Returns:

Self

__getitem__

fn def __getitem__(ref self, i: Int, j: Int) -> ref[self_is_mut] Scalar[Type]

Args:

  • self (Self)
  • i (Int)
  • j (Int)

Returns:

ref[self_is_mut] Scalar[Type]

slice

fn def slice(self, i: Int, j: Int, ir: Int, jr: Int) -> Self

Args:

  • self (Self)
  • i (Int)
  • j (Int)
  • ir (Int)
  • jr (Int)

Returns:

Self

shape

fn def shape[dim: Int](self) -> Int

Parameters:

  • dim (Int)

Args:

  • self (Self)

Returns:

Int

stride

fn def stride[dim: Int](self) -> Int

Parameters:

  • dim (Int)

Args:

  • self (Self)

Returns:

Int

rand

fn def rand(mut self)

Args:

  • self (Self)

load

fn def load[width: Int, *, dim: Int](self, i: Int, j: Int) -> SIMD[Type, width]

Parameters:

  • width (Int)
  • dim (Int)

Args:

  • self (Self)
  • i (Int)
  • j (Int)

Returns:

SIMD[Type, width]

store

fn def store[width: Int, *, dim: Int](self, value: SIMD[Type, width], i: Int, j: Int)

Parameters:

  • width (Int)
  • dim (Int)

Args:

  • self (Self)
  • value (SIMD[Type, width])
  • i (Int)
  • j (Int)

write_to

fn def write_to[W: Writer](self, mut writer: W)

Parameters:

  • W (Writer)

Args:

  • self (Self)
  • writer (W)
Edit this page