softmaxBackward

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