huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 02:58:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add the topological sort for backprop.

Browse Source
This commit is contained in:
laurent
2023-06-20 19:15:39 +01:00
parent 671bcf060e
commit 9ff8d2076a
3 changed files with 58 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/shape.rs
View File

@ -131,7 +131,7 @@ mod tests {
#[test]
fn stride() {
let shape = Shape::from(());
assert_eq!(shape.stride_contiguous(), []);
assert_eq!(shape.stride_contiguous(), Vec::<usize>::new());
let shape = Shape::from(42);
assert_eq!(shape.stride_contiguous(), [1]);
let shape = Shape::from((42, 1337));