Fork me on GitHub
Arraymancer Technical reference Tutorial Spellbook (How-To's) Under the hood

Module nnp_linear

Search:
Group by:

Procs

proc linear[T](input, weight: Tensor[T]; bias: Tensor[T]; output: var Tensor[T]) {.
inline
.}
  Source Edit
proc linear[T](input, weight: Tensor[T]; output: var Tensor[T]) {.
inline
.}
  Source Edit
proc linear_backward[T](input, weight, bias, gradOutput: Tensor[T];
                       gradInput, gradWeight, gradBias: var Tensor[T]) {.
inline
.}
  Source Edit
proc linear_backward[T](input, weight, gradOutput: Tensor[T];
                       gradInput, gradWeight: var Tensor[T]) {.
inline
.}
  Source Edit