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