Fork me on GitHub

src/arraymancer/nn/loss/cross_entropy_losses

Search:
Group by:
  Source Edit

Types

SigmoidCrossEntropyLoss[TT] {.inject, final.} = ref object of Gate[TT]
  Source Edit 
SoftmaxCrossEntropyLoss[TT] {.inject, final.} = ref object of Gate[TT]
  Source Edit 
SparseSoftmaxCrossEntropyLoss[TT; Idx] {.final.} = ref object of Gate[TT]
  Source Edit

Procs

proc sigmoid_cross_entropy[TT](a`gensym0: Variable[TT]; target`gensym0: TT): Variable[
    TT]
  Source Edit 
proc softmax_cross_entropy[TT](a`gensym1: Variable[TT]; target`gensym1: TT): Variable[
    TT]
  Source Edit 
proc sparse_softmax_cross_entropy[TT; Idx: SomeNumber or byte or char or enum](
    a: Variable[TT]; target: Tensor[Idx]): Variable[TT]
  Source Edit

Made with Nim. Generated: 2024-05-12 14:01:07 UTC
Arraymancer Technical reference Tutorial Spellbook (How-To's) Under the hood