Pow

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, DeviceStorage) _x): 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.

Mixins

__anonymous mixin FunctionCommon: Undocumented in source.

Variables

dx Variable!(T, dim, DeviceStorage) dx;: Undocumented in source.
hx Variable!(T, dim, HostStorage) hx;: Undocumented in source.
power T power;: 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

import grain.testing;
import std.typecons;
import numir;
import mir.ndslice;
import mir.math : pow;

auto p = 2.0f;
auto hfunc = Pow!(float, 2)(p);
auto hx = uniform!float(2, 3).slice.variable;
auto hy = hfunc.forward(hx);
auto hgy = uniform!float(2, 3).slice.variable;
auto hgx = hfunc.backward(hgy);
gradCheck(hfunc, hx, hgy, 1e-3, 1e-3, 1e-3);
assert(approxEqual(hy.sliced, hx.sliced.map!(a => pow(a, p))));

version (grain_cuda) {
    auto dfunc = Pow!(float, 2)(p);
    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));
}

Pow

Constructors

Members

Functions

Mixins

Variables

Mixed In Members

From `mixin FunctionCommon`

Examples

Meta

Source

Pow

Constructors

Members

Functions

Mixins

Variables

Mixed In Members

From mixin FunctionCommon

Examples

Meta

Source

From `mixin FunctionCommon`