grain ~master (2021-05-15T05:53:16Z)
Dub
Repo
ReLU.hx
grain
functions
unary
ReLU
Undocumented in source.
struct
ReLU
(T, size_t dim)
Variable
!(
T
,
dim
,
HostStorage
)
hx
;
Meta
Source
See Implementation
grain
functions
unary
ReLU
functions
backward
forward
mixins
__anonymous
variables
dx
dy
hx
inplace
useCuDNN