Ben_Kosytorz
3a67c0979c
- Updated stablediffusion crate path from "../stable-diffusion-burn" to "./crates/stable-diffusion-burn" for proper workspace resolution - Enhanced .gitignore to include generated model files (.mpk, .pt, .bin, .safetensors, .ckpt) and user_data directory - Added Cargo.lock to gitignore with appropriate comment - Reorganized IDE files section in gitignore for better clarity - Added newline at end of file for proper formatting
46 lines
1.3 KiB
Rust
46 lines
1.3 KiB
Rust
use super::*;
|
|
use burn_tensor::{Int, Tensor, TensorData};
|
|
|
|
#[test]
|
|
fn should_cast_int_to_float() {
|
|
const START: usize = 0;
|
|
const END: usize = 100;
|
|
|
|
let device = Default::default();
|
|
let tensor = Tensor::<TestBackend, 1, Int>::arange(START as i64..END as i64, &device);
|
|
|
|
let data_int = tensor.to_data();
|
|
let data_int = data_int.as_slice::<i32>().unwrap();
|
|
let data_float = tensor.float().into_data();
|
|
let data_float = data_float.as_slice::<f32>().unwrap();
|
|
|
|
for i in START..END {
|
|
assert_eq!(data_int[i], i as i32);
|
|
assert_eq!(data_float[i], i as f32);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn should_cast_bool_to_int() {
|
|
let device = Default::default();
|
|
|
|
let tensor_1 = Tensor::<TestBackend, 2>::from_floats([[1., 0., 3.], [0., 0., 900.]], &device);
|
|
let tensor_2: Tensor<TestBackend, 2, Int> = tensor_1.clone().greater_elem(0.0).int();
|
|
|
|
tensor_2
|
|
.to_data()
|
|
.assert_eq(&TensorData::from([[1, 0, 1], [0, 0, 1]]), false);
|
|
}
|
|
|
|
#[test]
|
|
fn should_cast_bool_to_float() {
|
|
let device = Default::default();
|
|
|
|
let tensor_1 = Tensor::<TestBackend, 2>::from_floats([[1., 0., 3.], [0., 0., 900.]], &device);
|
|
let tensor_2: Tensor<TestBackend, 2> = tensor_1.clone().greater_elem(0.0).float();
|
|
|
|
tensor_2
|
|
.to_data()
|
|
.assert_eq(&TensorData::from([[1., 0., 1.], [0., 0., 1.]]), false);
|
|
}
|