Fork me on GitHub

src/arraymancer/nn_primitives/private/p_logsumexp

  Source Edit

Procs

proc logsumexp[T: SomeFloat](t: Tensor[T]): T
  Source Edit
proc stable_softmax[T](x, max, sumexp: T): T {.noSideEffect, inline.}
  Source Edit
proc streaming_max_sumexp[T](t: Tensor[T]): tuple[max: T, sumexp: T] {.
    noSideEffect, inline.}
  Source Edit
proc streaming_max_sumexp[T](t: Tensor[T]; axis: int): Tensor[
    tuple[max: T, sumexp: T]] {.noinit.}
  Source Edit
Arraymancer Technical reference Tutorial Spellbook (How-To's) Under the hood