pub fn reduce<Run: CubeRuntime>(
tensor: CubeTensor<Run>,
output_dtype: Option<DType>,
strategy: KernelReduceStrategy,
config: ReduceOperationConfig,
) -> Result<CubeTensor<Run>, ReduceError>Expand description
Reduce all elements of the input tensor using the instruction Rd and the given Strategy.
Return an error if strategy is Specific(strategy) and the specified strategy is not supported by the client.
If there is no error, the output is a tensor with decreasing strides where the shape of reduced dim is set to 1 but all shape are similar to the input.