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

Updated everything and output a trace.

Browse Source
This commit is contained in:
Nicolas Patry
2023-11-07 21:06:20 +01:00
parent 76d3116f5d
commit 0c24a885a6
8 changed files with 249 additions and 88 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
candle-nn/src/ops.rs
View File

@ -1,5 +1,6 @@
use candle::{CpuStorage, Layout, Result, Shape, Tensor};
use rayon::prelude::*;
use tracing::debug;
/// Applies the softmax function to the input tensor, rescaling the element so that elements on
/// a slice of fixed index on dimension `dim` are between 0 and 1 and sum to 1.
@ -198,7 +199,7 @@ impl candle::CustomOp1 for SoftmaxLastDim {
storage: &candle::MetalStorage,
layout: &Layout,
) -> Result<(candle::MetalStorage, Shape)> {
println!("TODO softmax-last-dim");
debug!("TODO softmax-last-dim");
Ok((storage.clone(), layout.shape().clone()))
}
}