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

Skeleton implementation for softmax.

Browse Source
This commit is contained in:
laurent
2023-06-23 22:00:13 +01:00
parent 5d44e76e3f
commit 8443963d4f
5 changed files with 32 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/cpu_backend.rs
View File

@ -147,6 +147,10 @@ impl CpuStorage {
}
}
pub(crate) fn divide_by_sum_over_dim(&mut self, _shape: &Shape, _dim: usize) {
todo!()
}
pub(crate) fn affine_impl(
&self,
shape: &Shape,