Reduction ops
Generated from the binaries by scripts/gen-ops-reference.ts — do not edit by
hand. Every op also documents itself: torch <op> --help.
sum
sum over all elements, or along —dim
torch sum <t1> [--dim <Int>] [--keepdim]torch sum <t1> [--dim <Int>] [--keepdim]mean
mean over all elements, or along —dim (float32, v1 fidelity)
torch mean <t1> [--dim <Int>] [--keepdim]torch mean <t1> [--dim <Int>] [--keepdim]prod
product over all elements, or along —dim
torch prod <t1> [--dim <Int>] [--keepdim]torch prod <t1> [--dim <Int>] [--keepdim]amax
max values over all elements, or along —dim
torch amax <t1> [--dim <Int>] [--keepdim]torch amax <t1> [--dim <Int>] [--keepdim]amin
min values over all elements, or along —dim
torch amin <t1> [--dim <Int>] [--keepdim]torch amin <t1> [--dim <Int>] [--keepdim]max
max of all elements; with —dim also returns indices
torch max <t1> [--dim <Int>] [--keepdim]torch max <t1> [--dim <Int>] [--keepdim]min
min of all elements; with —dim also returns indices
torch min <t1> [--dim <Int>] [--keepdim]torch min <t1> [--dim <Int>] [--keepdim]median
median of all elements; with —dim also returns indices
torch median <t1> [--dim <Int>] [--keepdim]torch median <t1> [--dim <Int>] [--keepdim]argmax
index of the max, overall or along —dim
torch argmax <t1> [--dim <Int>] [--keepdim]torch argmax <t1> [--dim <Int>] [--keepdim]argmin
index of the min, overall or along —dim
torch argmin <t1> [--dim <Int>] [--keepdim]torch argmin <t1> [--dim <Int>] [--keepdim]all
true if all elements are true (Bool tensor)
torch all <t1> [--dim <Int>] [--keepdim]torch all <t1> [--dim <Int>] [--keepdim]any
true if any element is true (Bool tensor)
torch any <t1> [--dim <Int>] [--keepdim]torch any <t1> [--dim <Int>] [--keepdim]std
standard deviation (—correction, default 1)
torch std <t1> [--dim <Int>] [--keepdim] [--correction <Int>]torch std <t1> [--dim <Int>] [--keepdim] [--correction <Int>]var
variance (—correction, default 1)
torch var <t1> [--dim <Int>] [--keepdim] [--correction <Int>]torch var <t1> [--dim <Int>] [--keepdim] [--correction <Int>]nansum
sum treating NaN as zero
torch nansum <t1> [--dim <Int>] [--keepdim]torch nansum <t1> [--dim <Int>] [--keepdim]logsumexp
log(sum(exp(x))) along —dim
torch logsumexp <t1> [--dim <Int>] [--keepdim]torch logsumexp <t1> [--dim <Int>] [--keepdim]count_nonzero
count of nonzero elements, overall or along —dim
torch count_nonzero <t1> [--dim <Int>]torch count_nonzero <t1> [--dim <Int>]cumsum
cumulative sum along —dim
torch cumsum <t1> [--dim <Int>]torch cumsum <t1> [--dim <Int>]cumprod
cumulative product along —dim
torch cumprod <t1> [--dim <Int>]torch cumprod <t1> [--dim <Int>]norm
p-norm (—p default 2), overall or along —dim
torch norm <t1> [--p <Float>] [--dim <Int>] [--keepdim]torch norm <t1> [--p <Float>] [--dim <Int>] [--keepdim]diff
forward differences along —dim (default last)
torch diff <t1> [--dim <Int>]torch diff <t1> [--dim <Int>]