diff options
| author | Nicolas Patry <patry.nicolas@protonmail.com> | 2023-12-18 11:04:16 +0100 |
|---|---|---|
| committer | Nicolas Patry <patry.nicolas@protonmail.com> | 2023-12-18 11:04:16 +0100 |
| commit | 064ba17bd7de0e3c6f18f93cfe14db97e7ebca0b (patch) | |
| tree | 07cd67136902a4ccb07bdf11bf9961ec9eef1e92 | |
| parent | e8ee253ee0766c33ac69f08bb0bcd6601f47ca6f (diff) | |
| download | candle-064ba17bd7de0e3c6f18f93cfe14db97e7ebca0b.tar.gz candle-064ba17bd7de0e3c6f18f93cfe14db97e7ebca0b.tar.bz2 candle-064ba17bd7de0e3c6f18f93cfe14db97e7ebca0b.zip | |
Remove print.
| -rw-r--r-- | candle-core/tests/tensor_tests.rs | 1 |
1 files changed, 0 insertions, 1 deletions
diff --git a/candle-core/tests/tensor_tests.rs b/candle-core/tests/tensor_tests.rs index 06891748..c871dc96 100644 --- a/candle-core/tests/tensor_tests.rs +++ b/candle-core/tests/tensor_tests.rs @@ -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)? |