huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 11:08:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

More general broadcast setup.

Browse Source
This commit is contained in:
laurent
2023-06-25 08:55:09 +01:00
parent 213445c0e5
commit 7ccf27dda2
3 changed files with 48 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/tensor_tests.rs
View File

@ -144,7 +144,7 @@ fn broadcast() -> Result<()> {
let data = &[3f32, 1., 4.];
let tensor = Tensor::new(data, &Device::Cpu)?;
assert_eq!(
tensor.broadcast((3, 1))?.to_vec3::<f32>()?,
tensor.broadcast_left((3, 1))?.to_vec3::<f32>()?,
&[[[3.0, 1.0, 4.0]], [[3.0, 1.0, 4.0]], [[3.0, 1.0, 4.0]]]
);
Ok(())