Fork me on GitHub

src/arraymancer/nn_primitives/nnp_linear

  Source Edit

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, 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
Arraymancer Technical reference Tutorial Spellbook (How-To's) Under the hood