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

Tweak some comment.

Browse Source
This commit is contained in:
laurent
2023-06-28 21:10:54 +01:00
parent 2ae368e98e
commit 46c07b924c
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/cpu_backend.rs
View File

@ -397,7 +397,7 @@ impl CpuStorage {
} }
pub(crate) fn divide_by_sum_over_dim(&mut self, shape: &Shape, dim: usize) -> Result<()> { pub(crate) fn divide_by_sum_over_dim(&mut self, shape: &Shape, dim: usize) -> Result<()> {
// [self] stores data in a contiguous way. // [self] stores data in a contiguous way starting at offset 0.
let dims = shape.dims(); let dims = shape.dims();
let elem_per_slice = dims[dim]; let elem_per_slice = dims[dim];
let prod_pre_dim = dims[..dim].iter().product(); let prod_pre_dim = dims[..dim].iter().product();