def take

bendtorch-core@0.1.0.0 · src/tensor/elementwise.bend:459

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

459def take(dynamic: Core.Dynamic) -> Core.Dynamic & Array<F32>:
460  Core.Packed{+shape, Core.TTensor{storage, exact}} = dynamic
461  match storage:
462    case Storage.Empty{}:
463      (Core.Packed{shape, Core.TTensor{Storage.Empty{}, exact}}, Array.new(F32, 0n, 0.0))
464    case Storage.Owned{+depth, +length, array, balanced, valid}:
465      take_owned(shape, depth, length, array, Array.clone(F32, array), {==}, balanced, valid, exact)