MCPcopy Create free account
hub / github.com/evilsocket/cake / sigmoid

Method sigmoid

cake-core/src/backends/mod.rs:508–510  ·  view source on GitHub ↗

Sigmoid activation: `1 / (1 + exp(-x))`.

(&self, x: &Tensor)

Source from the content-addressed store, hash-verified

506
507 /// Sigmoid activation: `1 / (1 + exp(-x))`.
508 fn sigmoid(&self, x: &Tensor) -> Result<Tensor> {
509 candle_nn::ops::sigmoid(x)
510 }
511
512 /// Embedding lookup: select rows from weight matrix by token IDs.
513 fn embedding(&self, ids: &Tensor, weight: &Tensor) -> Result<Tensor> {

Callers

nothing calls this directly

Implementers 5

mod.rscake-core/src/backends/rocm/mod.rs
mod.rscake-core/src/backends/cpu/mod.rs
mod.rscake-core/src/backends/metal/mod.rs
mod.rscake-core/src/backends/vulkan/mod.rs
mod.rscake-core/src/backends/cuda/mod.rs

Calls 1

sigmoidFunction · 0.85

Tested by

no test coverage detected