Neg

y = -x

struct Neg (

size_t dim

) {

import mir.ndslice : slice;;

auto forward(Variable!(T, dim, HostStorage) x);

auto backward(Variable!(T, dim, HostStorage) gy);

import grain.kernel : neg;;

version(grain_cuda)

auto forward(Variable!(T, dim, DeviceStorage) x);

version(grain_cuda)

auto backward(Variable!(T, dim, DeviceStorage) gy);

mixin FunctionCommon ;

}

Members

Functions

backward auto backward(Variable!(T, dim, HostStorage) gy): Undocumented in source. Be warned that the author may not have intended to support it.
backward auto backward(Variable!(T, dim, DeviceStorage) gy): Undocumented in source. Be warned that the author may not have intended to support it.
forward auto forward(Variable!(T, dim, HostStorage) x): Undocumented in source. Be warned that the author may not have intended to support it.
forward auto forward(Variable!(T, dim, DeviceStorage) x): Undocumented in source. Be warned that the author may not have intended to support it.

Mixins

__anonymous mixin FunctionCommon: Undocumented in source.

Mixed In Members

From `mixin FunctionCommon`

this(this) this(this): Undocumented in source.
DeviceRets alias DeviceRets = Tuple!(Parameters!backward): Undocumented in source.
DeviceArgs alias DeviceArgs = Tuple!(Parameters!forward): Undocumented in source.
__anonymous mixin TypeChecker!(forward, backward): Undocumented in source.
_mixin_dargs DeviceArgs _mixin_dargs;: Undocumented in source.
HostRets alias HostRets = Tuple!(Parameters!backward): Undocumented in source.
HostArgs alias HostArgs = Tuple!(Parameters!forward): Undocumented in source.
__anonymous mixin TypeChecker!(forward, backward): Undocumented in source.
_mixin_hargs HostArgs _mixin_hargs;: Undocumented in source.
applyForward auto applyForward(Args args): store grain.autograd.BackProp object in returned variables from forward function
applyBackward void applyBackward(UntypedVariable[] ugradOutputs): type-erased version of backward function used in grain.autograd.BackProp object

Examples

test neg simple case, gradcheck and cpu/cuda equality

import grain.testing;
import std.typecons;
import numir;
import mir.ndslice;

// simple case: 2.0 * x
auto xs = [[-1.0f, 2.0f, 3.0f], [1.0f, 0.1f, 0.0f]].nparray;
auto hfunc = Neg!(float, 2)();
auto _hx = xs.variable;
auto _hy = hfunc.forward(_hx);
assert(approxEqual(_hy.sliced, -xs));

auto hx = uniform!float(2, 2).slice.variable;
auto hy = hfunc.forward(hx);
auto hgy = uniform!float(2, 2).slice.variable;
auto hgx = hfunc.backward(hgy);
gradCheck(hfunc, hx, hgy); // , 1e-3, 1e-3, 1e-3);

version (grain_cuda) {
    auto dfunc = Neg!(float, 2)();
    auto dy = dfunc.forward(hx.to!DeviceStorage);
    assert(approxEqual(dy.to!HostStorage.sliced, hy.sliced));
    auto dgx = dfunc.backward(hgy.to!DeviceStorage);
    assert(approxEqual(dgx.to!HostStorage.sliced, hgx.sliced));
}

Neg

Members

Functions

Mixins

Mixed In Members

From `mixin FunctionCommon`

Examples

Meta

Source

Neg

Members

Functions

Mixins

Mixed In Members

From mixin FunctionCommon

Examples

Meta

Source

From `mixin FunctionCommon`