Documentation

softmaxBackward

graincudnn

grad of softmax

version(grain_cuda)
void
softmaxBackward
(
cudnnSoftmaxAlgorithm_t A
T
size_t dim
)
(
Variable!(T, dim, DeviceStorage) gx
,
Variable!(T, dim, DeviceStorage) gy
,
Variable!(T, dim, DeviceStorage) y
,
T alpha = 1.0
,
T beta = 0.0
)

Meta

Source

See Implementation

graincudnn functionsstructs
Distributed under BSL-1.0. Copyright Shigeki Karita 2018. Page generated by adrdox