Scan/Sort Ops#
Applies the combine_fn to each elements with a carry in |
|
Returns the cumprod of all elements in the |
|
Returns the cumsum of all elements in the |
|
computes an histogram based on input tensor with num_bins bins, the bins have a width of 1 and start at 0. |
|
Returns the k largest (or smallest) elements of the input tensor along the specified dimension. |