Scan/Sort Ops

Scan/Sort Ops#

associative_scan

Applies the combine_fn to each elements with a carry in input tensors along the provided axis and update the carry

bitonic_merge

cumprod

Returns the cumprod of all elements in the input tensor along the provided axis

cumsum

Returns the cumsum of all elements in the input tensor along the provided axis

histogram

computes an histogram based on input tensor with num_bins bins, the bins have a width of 1 and start at 0.

sort

topk

Returns the k largest (or smallest) elements of the input tensor along the specified dimension.