grain ~master (2021-05-15T05:53:16Z)
Dub
Repo
LogSoftmax.dy
grain
functions
unary
LogSoftmax
Undocumented in source.
struct
LogSoftmax
(T, size_t dim = 2)
version(grain_cuda)
Variable
!(
T
,
dim
,
DeviceStorage
)
dy
;
Meta
Source
See Implementation
grain
functions
unary
LogSoftmax
functions
backward
forward
mixins
__anonymous
variables
dy
hy