burn-tensor 0.22.0

Tensor library with user-friendly APIs and automatic differentiation support
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

//! A separate test binary starts with uninitialized device settings.

use burn_dispatch::devices::FlexDevice;
use burn_tensor::{DType, Device, DeviceError, FloatDType, Tensor};

#[test]
fn explicit_tensor_creation_locks_default_settings() {
    // Flex supports F64 regardless of which other backends are enabled.
    let mut device = Device::new(FlexDevice);
    let tensor = Tensor::<1>::from_data([1.0f64], (&device, DType::F64));

    assert_eq!(tensor.dtype(), DType::F64);
    assert_eq!(device.settings().float_dtype, FloatDType::F32);
    assert!(matches!(
        device.configure(FloatDType::F64),
        Err(DeviceError::AlreadyInitialized { .. })
    ));
    assert_eq!(Tensor::<1>::zeros([1], &device).dtype(), DType::F32);
}