Documentation

Init.Data.FloatArray.Basic

structure FloatArray :

data : Array Float

Instances For

@[extern lean_mk_empty_float_array]

def FloatArray.mkEmpty (c : Nat) :

Equations

FloatArray.mkEmpty c = { data := #[] }

def FloatArray.empty :

Equations

FloatArray.empty = FloatArray.mkEmpty 0

instance FloatArray.instInhabited :

Inhabited FloatArray

Equations

FloatArray.instInhabited = { default := FloatArray.empty }

instance FloatArray.instEmptyCollection :

EmptyCollection FloatArray

Equations

FloatArray.instEmptyCollection = { emptyCollection := FloatArray.empty }

@[extern lean_float_array_push]

def FloatArray.push :

FloatArray → Float → FloatArray

Equations

{ data := ds }.push x✝ = { data := ds.push x✝ }

@[extern lean_float_array_size]

def FloatArray.size :

FloatArray → Nat

Equations

{ data := ds }.size = ds.size

@[extern lean_sarray_size]

def FloatArray.usize (a : FloatArray) :

Equations

a.usize = a.size.toUSize

@[extern lean_float_array_uget]

def FloatArray.uget (a : FloatArray) (i : USize) :

i.toNat < a.size → Float

Equations

{ data := ds }.uget x✝¹ x✝ = ds[x✝¹]

@[extern lean_float_array_fget]

def FloatArray.get (ds : FloatArray) (i : Nat) (h : i < ds.size := by get_elem_tactic) :

Equations

{ data := ds }.get x✝¹ x✝ = ds.get x✝¹ x✝

@[extern lean_float_array_get]

def FloatArray.get! :

FloatArray → Nat → Float

Equations

{ data := ds }.get! x✝ = ds.get! x✝

def FloatArray.get? (ds : FloatArray) (i : Nat) :

Equations

ds.get? i = if h : i < ds.size then some (ds.get i h) else none

instance FloatArray.instGetElemNatFloatLtSize :

GetElem FloatArray Nat Float fun (xs : FloatArray) (i : Nat) => i < xs.size

Equations

FloatArray.instGetElemNatFloatLtSize = { getElem := fun (xs : FloatArray) (i : Nat) (h : i < xs.size) => xs.get i h }

instance FloatArray.instGetElemUSizeFloatLtNatValValSize :

GetElem FloatArray USize Float fun (xs : FloatArray) (i : USize) => ↑i.val < xs.size

Equations

FloatArray.instGetElemUSizeFloatLtNatValValSize = { getElem := fun (xs : FloatArray) (i : USize) (h : ↑i.val < xs.size) => xs.uget i h }

@[extern lean_float_array_uset]

def FloatArray.uset (a : FloatArray) (i : USize) :

Float → (h : autoParam (i.toNat < a.size) _auto✝) → FloatArray

Equations

{ data := ds }.uset x✝² x✝¹ x✝ = { data := ds.uset x✝² x✝¹ x✝ }

@[extern lean_float_array_fset]

def FloatArray.set (ds : FloatArray) (i : Nat) :

Float → (h : autoParam (i < ds.size) _auto✝) → FloatArray

Equations

{ data := ds }.set x✝² x✝¹ x✝ = { data := ds.set x✝² x✝¹ x✝ }

@[extern lean_float_array_set]

def FloatArray.set! :

FloatArray → Nat → Float → FloatArray

Equations

{ data := ds }.set! x✝¹ x✝ = { data := ds.set! x✝¹ x✝ }

def FloatArray.isEmpty (s : FloatArray) :

Equations

s.isEmpty = (s.size == 0)

def FloatArray.toList (ds : FloatArray) :

Equations

ds.toList = FloatArray.toList.loop ds 0 []

partial def FloatArray.toList.loop (ds : FloatArray) (i : Nat) (r : List Float) :

@[inline]

unsafe def FloatArray.forInUnsafe {β : Type v} {m : Type v → Type w} [Monad m] (as : FloatArray) (b : β) (f : Float → β → m (ForInStep β)) :

m β

We claim this unsafe implementation is correct because an array cannot have more than usizeSz elements in our runtime. This is similar to the Array version.

Equations

as.forInUnsafe b f = FloatArray.forInUnsafe.loop as f as.usize 0 b

@[specialize #[]]

unsafe def FloatArray.forInUnsafe.loop {β : Type v} {m : Type v → Type w} [Monad m] (as : FloatArray) (f : Float → β → m (ForInStep β)) (sz i : USize) (b : β) :

m β

Equations

One or more equations did not get rendered due to their size.

@[implemented_by FloatArray.forInUnsafe]

def FloatArray.forIn {β : Type v} {m : Type v → Type w} [Monad m] (as : FloatArray) (b : β) (f : Float → β → m (ForInStep β)) :

m β

Reference implementation for forIn

Equations

as.forIn b f = FloatArray.forIn.loop as f as.size ⋯ b

def FloatArray.forIn.loop {β : Type v} {m : Type v → Type w} [Monad m] (as : FloatArray) (f : Float → β → m (ForInStep β)) (i : Nat) (h : i ≤ as.size) (b : β) :

m β

Equations

One or more equations did not get rendered due to their size.
FloatArray.forIn.loop as f 0 x b = pure b

instance FloatArray.instForInFloat {m : Type u_1 → Type u_2} :

ForIn m FloatArray Float

Equations

FloatArray.instForInFloat = { forIn := fun {β : Type ?u.15} [Monad m] => FloatArray.forIn }

@[inline]

unsafe def FloatArray.foldlMUnsafe {β : Type v} {m : Type v → Type w} [Monad m] (f : β → Float → m β) (init : β) (as : FloatArray) (start : Nat := 0) (stop : Nat := as.size) :

m β

See comment at forInUnsafe

Equations

One or more equations did not get rendered due to their size.

@[specialize #[]]

unsafe def FloatArray.foldlMUnsafe.fold {β : Type v} {m : Type v → Type w} [Monad m] (f : β → Float → m β) (as : FloatArray) (i stop : USize) (b : β) :

m β

Equations

FloatArray.foldlMUnsafe.fold f as i stop b = if (i == stop) = true then pure b else do let __do_lift ← f b (as.uget i ⋯) FloatArray.foldlMUnsafe.fold f as (i + 1) stop __do_lift

@[implemented_by FloatArray.foldlMUnsafe]

def FloatArray.foldlM {β : Type v} {m : Type v → Type w} [Monad m] (f : β → Float → m β) (init : β) (as : FloatArray) (start : Nat := 0) (stop : Nat := as.size) :

m β

Reference implementation for foldlM

Equations

One or more equations did not get rendered due to their size.

def FloatArray.foldlM.loop {β : Type v} {m : Type v → Type w} [Monad m] (f : β → Float → m β) (as : FloatArray) (stop : Nat) (h : stop ≤ as.size) (i j : Nat) (b : β) :

m β

Equations

One or more equations did not get rendered due to their size.

@[inline]

def FloatArray.foldl {β : Type v} (f : β → Float → β) (init : β) (as : FloatArray) (start : Nat := 0) (stop : Nat := as.size) :

β

Equations

FloatArray.foldl f init as start stop = (FloatArray.foldlM f init as start stop).run

def List.toFloatArray (ds : List Float) :

Equations

ds.toFloatArray = List.toFloatArray.loop ds FloatArray.empty

def List.toFloatArray.loop :

List Float → FloatArray → FloatArray

Equations

List.toFloatArray.loop [] x✝ = x✝
List.toFloatArray.loop (b :: ds) x✝ = List.toFloatArray.loop ds (x✝.push b)

instance instToStringFloatArray :

ToString FloatArray

Equations

instToStringFloatArray = { toString := fun (ds : FloatArray) => ds.toList.toString }