Sum

Members

Functions

backward

auto backward(Variable!(T, 0, HostStorage) y)

Undocumented in source. Be warned that the author may not have intended to support it.

backward

auto backward(Variable!(T, 0, DeviceStorage) y)

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.

Variables

shape

uint[dim] shape;

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

import mir.ndslice;
import mir.math;
auto x = [1f, 2f, 3f, 4f].sliced(2, 2).variable;
Sum!("fast", float, 2) func;
auto y = func.forward(x);
assert(y == 10f.variable);
assert(func.backward(1.2f.variable) == [1.2f, 1.2f, 1.2f, 1.2f].sliced(2, 2).variable);

version (grain_cuda) {
    auto cx = x.to!DeviceStorage;
    auto cy = func.forward(cx).to!HostStorage;
    assert(cy == 10f.variable);
    auto cgx = func.backward(1.2f.variable.to!DeviceStorage).to!HostStorage;
    assert(cgx.sliced == [1.2f, 1.2f, 1.2f, 1.2f].sliced(2, 2));
}

• Sum