def set_length

bendtorch-core@0.1.0.0 · src/tensor/lifecycle.bend:57

import Base
import bendtorch-core@0.1.0.0/src/storage/main.bend as Storage

57def set_length(storage: Storage.Storage, +i: U32, +value: F32) ->
58  {retained_write(Storage.set(storage, i, value)) == logical_length(storage) : Nat}:
59  match storage:
60    case Storage.Empty{}:
61      {==}
62    case Storage.Owned{+d, +n, array, balanced, valid}:
63      set_checked_length(d, n, array, i, value, Nat.is_lt(U32.to_nat(i), n), {==}, balanced, valid)