golem ~master (2021-09-25T16:54:00Z)
Dub
Repo
Linear.opCall
golem
nn
Linear
Undocumented in source. Be warned that the author may not have intended to support it.
class
Linear
(T, size_t InputDim, size_t OutputDim, UseGradient useGradient = UseGradient.yes)
auto
opCall
(
size_t
[2]
Shape
UseGradient
useGradient
)
(
Tensor
!(
T
,
Shape
,
useGradient
)
x
)
if
(
Shape
[1] ==
InputDim
)
Meta
Source
See Implementation
golem
nn
Linear
aliases
parameters
constructors
this
functions
opCall
resetGrads
variables
bias
weights