TensorCreateOptions

import { TensorCreateOptions } from "destack:tensor";

Allocation options for a new tensor.

export struct TensorCreateOptions<const out Rank: usize, in out F: TensorFormat = Dense<MemoryOrder.RowMajor>, P: Placement = Unsharded> {
    placement: P;
    device: Device;
    mesh: Mesh;
    stream: Stream;
}
language/library/src/tensor/options.ds:16:32

Members

placement: P

Tensor placement.

device: Device

Device used for unsharded placement.

mesh: Mesh

Mesh used for sharded placement.

stream: Stream

Stream used to order initialization work.