Struct TensorCreationOptions 

pub struct TensorCreationOptions {
    pub device: Device,
    pub dtype: Option<DType>,
}

Options for tensor creation.

This struct allows specifying the device and overriding the data type when creating a tensor. When the dtype is not specified, the device’s default settings is used.

Fields

device: Device

Device where the tensor will be created.

dtype: Option<DType>

Optional data type. If None, the dtype will be inferred on creation from the device settings.

Implementations

impl TensorCreationOptions

pub fn new(device: Device) -> TensorCreationOptions

Create new options with a specific device.

Data type will follow the device settings on tensor creation.

pub fn with_dtype(self, dtype: DType) -> TensorCreationOptions

Set the tensor creation data type.

pub fn with_device(self, device: Device) -> TensorCreationOptions

Set the tensor creation device.

pub fn dtype_or(&self, dtype: DType) -> DType

Returns the tensor data type, or a provided default if not set.

This is useful for cases where TensorCreationOptions may not have an explicit dtype.

Trait Implementations

impl Clone for TensorCreationOptions

fn clone(&self) -> TensorCreationOptions

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for TensorCreationOptions

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.

impl Default for TensorCreationOptions

fn default() -> TensorCreationOptions

Returns new options with the backend’s default device.

impl From<&Device> for TensorCreationOptions

fn from(device: &Device) -> TensorCreationOptions

Convenience conversion from a reference to a device.

Example:

use burn_tensor::TensorCreationOptions;
use burn_tensor::Device;

fn example(device: Device) {
    let options: TensorCreationOptions = (&device).into();
}

impl From<(&Device, DType)> for TensorCreationOptions

fn from(args: (&Device, DType)) -> TensorCreationOptions

Convenience conversion for a specified (&device, dtype) tuple.