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