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

ArgMin now returns a tensor with i64

Browse Source
This commit is contained in:
b1rtek
2024-05-09 20:22:22 +02:00
parent 13b88547f7
commit 9a273196b7
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-onnx/src/eval.rs
View File

@ -1028,7 +1028,7 @@ pub fn simple_eval(
input.argmin_keepdim(axis)?
} else {
input.argmin(axis)?
};
}.to_dtype(DType::I64)?;
values.insert(node.output[0].clone(), output);
}
op_type => bail!("unsupported op_type {op_type} for op {node:?}"),