mirror of
https://github.com/huggingface/candle.git
synced 2025-06-16 18:48:51 +00:00
Add more to the binary operators.
This commit is contained in:
@ -144,6 +144,7 @@ impl Shape {
|
||||
pub(crate) fn stride_contiguous(&self) -> Vec<usize> {
|
||||
self.0
|
||||
.iter()
|
||||
.rev()
|
||||
.scan(1, |prod, u| {
|
||||
let prod_pre_mult = *prod;
|
||||
*prod *= u;
|
||||
|
Reference in New Issue
Block a user