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

Bugfix for to_scalar (use the proper start offset).

Browse Source
This commit is contained in:
laurent
2023-07-05 06:42:29 +01:00
parent ad0666271d
commit 6d1e79d378
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-core/src/tensor.rs
View File

@ -326,7 +326,7 @@ impl Tensor {
}
let from_cpu_storage = |cpu_storage: &crate::CpuStorage| {
let data = S::cpu_storage_as_slice(cpu_storage)?;
Ok::<_, Error>(data[0])
Ok::<_, Error>(data[self.layout().start_offset()])
};
match self.storage.as_ref() {
Storage::Cpu(cpu_storage) => from_cpu_storage(cpu_storage),