Struct TensorCreationOptions 

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

Options for tensor creation.

This struct allows specifying the device and/or data type (dtype) when creating a tensor.

Fields

device: Device<B>

Device where the tensor will be created.

dtype: Option<DType>

Optional data type. If None, the dtype will be inferred on creation from the backend’s default dtype for the tensor kind.

Implementations

impl<B: Backend> TensorCreationOptions<B>

pub fn new(device: Device<B>) -> Self

Create new options with a specific device.

Data type will be inferred on creation from the backend’s default dtype for the tensor kind.

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

Set the tensor creation data type.

pub fn with_device(self, device: Device<B>) -> Self

Set the tensor creation device.

pub fn float() -> Self

Create options with backend’s default device and float dtype.

pub fn int() -> Self

Create options with backend’s default device and int dtype.

pub fn bool() -> Self

Create options with backend’s default device and bool dtype.

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<B: Clone + Backend> Clone for TensorCreationOptions<B>

fn clone(&self) -> TensorCreationOptions<B>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<B: Debug + Backend> Debug for TensorCreationOptions<B>

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

Formats the value using the given formatter.

impl<B: Backend> Default for TensorCreationOptions<B>

fn default() -> Self

Returns new options with the backend’s default device.

impl<B: Backend> From<&<B as Backend>::Device> for TensorCreationOptions<B>

fn from(device: &Device<B>) -> Self

Convenience conversion from a reference to a device.

Example:

use burn_tensor::backend::Backend;
use burn_tensor::TensorCreationOptions;

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

impl<B: Backend> From<(&<B as Backend>::Device, DType)> for TensorCreationOptions<B>

fn from(args: (&Device<B>, DType)) -> Self

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