grain ~master (2021-05-15T05:53:16Z)

ReLU.dx

grain functions unary ReLU
Undocumented in source.
  1. Variable!(T, dim, DeviceStorage) dx;
    struct ReLU(T, size_t dim)
    version(grain_cuda)
    Variable!(T, dim, DeviceStorage) dx;
  2. Variable!(T, dim, DeviceStorage) dy;

Meta

Source

See Implementation

grain functions unary ReLU functionsmixinsvariables

Page generated by adrdox