def select_chosen

stelliferous@0.0.2.0 · tensor_lanes.bend:114

import Base
import stelliferous@0.0.2.0/math_compare.bend as Compare
import stelliferous@0.0.2.0/storage_buffer.bend as Storage
import stelliferous@0.0.2.0/tensor_operations.bend as Operations
import stelliferous@0.0.2.0/tensor_view.bend as Views

114def select_chosen(+count: Nat,+position: U32,no: Storage.Buffer<F32>,result: Storage.Buffer<F32> & Storage.Buffer<F32>) -> Selected:
115  (yes,mask) = result
116  select_rejected(count,position,yes,
117    Operations.combine_kernel(~F32,~Compare.rejected,count,Views.View{[Views.Axis{U32.from_nat(count),1}],position},position,no,mask))