grain.cudnn

Members

Functions

activationBackward
void activationBackward(Variable!(T, dim, DeviceStorage) gx, Variable!(T, dim, DeviceStorage) gy, Variable!(T, dim, DeviceStorage) x, Variable!(T, dim, DeviceStorage) y, T alpha = 1.0, T beta = 0.0, double coeff = 0.0)
activationForward
void activationForward(Variable!(T, dim, DeviceStorage) x, Variable!(T, dim, DeviceStorage) y, T alpha = 1.0, T beta = 0.0, double coeff = 0.0)

y = alpha * f(x) + beta * y

softmaxForward
void softmaxForward(Variable!(T, dim, DeviceStorage) x, Variable!(T, dim, DeviceStorage) y, T alpha = 1.0, T beta = 0.0)

compute the softmax over all C for each H, W, N

Meta