huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove print.

Browse Source
This commit is contained in:
Nicolas Patry
2023-12-18 11:04:16 +01:00
parent e8ee253ee0
commit 064ba17bd7
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
candle-core/tests/tensor_tests.rs
View File

@ -543,7 +543,6 @@ fn argmax(device: &Device) -> Result<()> {
let t1 = tensor.reshape((190, 5, 4))?;
let t2 = t1.transpose(0, 2)?.contiguous()?.transpose(0, 2)?;
for tensor in [t1, t2] {
println!("{}", tensor.argmax_keepdim(0)?.argmax_keepdim(2)?);
assert_eq!(
tensor
.argmax_keepdim(0)?