grain ~master (2021-05-15T05:53:16Z)
Dub
Repo
NegativeLogLikelihood.forward
grain
functions
loss
NegativeLogLikelihood
Undocumented in source. Be warned that the author may not have intended to support it.
auto
forward
(Variable!(F, 2, HostStorage) logP, Variable!(I, 1, HostStorage) targetId)
auto
forward
(Variable!(F, 2, DeviceStorage) logP, Variable!(I, 1, DeviceStorage) targetId)
struct
NegativeLogLikelihood
(F, I = long)
version(grain_cuda)
auto
forward
(
Variable
!(
F
, 2,
DeviceStorage
)
logP
,
Variable
!(
I
, 1,
DeviceStorage
)
targetId
)
Meta
Source
See Implementation
grain
functions
loss
NegativeLogLikelihood
functions
backward
forward
mixins
__anonymous
variables
_dtargetId
_htargetId
_nClass
_normalize
ignoreIndex
sizeAverage