TensorReduceOptions

import { TensorReduceOptions } from "destack:tensor";

Output options for tensor reductions.

export struct TensorReduceOptions<T, const Rank: usize, F: TensorFormat = Dense<MemoryOrder.RowMajor>, P: Placement = Unsharded> {
    keepDims: boolean;
    placement: P;
    device: Device;
    mesh: Mesh;
    stream: Stream;
    out: TensorView<T, Rank, F, P, "exclusive">;
}
language/library/src/tensor/options.ds:58:81

Members

keepDims: boolean

Whether reduced axes remain with extent one.

placement: P

Result placement.

device: Device

Device used for unsharded placement.

mesh: Mesh

Mesh used for sharded placement.

stream: Stream

Stream used to order the work.

out: TensorView<T, Rank, F, P, "exclusive">

Exclusive output view to write into.