grain ~master (2021-05-15T05:53:16Z)
Dub
Repo
ReLU.forward
grain
functions
unary
ReLU
Undocumented in source. Be warned that the author may not have intended to support it.
auto
forward
(Variable!(T, dim, HostStorage) x)
struct
ReLU
(T, size_t dim)
auto
forward
(
Variable
!(
T
,
dim
,
HostStorage
)
x
)
auto
forward
(Variable!(T, dim, DeviceStorage) x)
Meta
Source
See Implementation
grain
functions
unary
ReLU
functions
backward
forward
mixins
__anonymous
variables
dx
dy
hx
inplace
useCuDNN