Pool

max pooling function

struct Pool (

bool isMax

T

size_t poolDims

) {

int[poolDims] window;

int[poolDims] pad;

int[poolDims] stride;

Variable!(T, dim, HostStorage) hx;

Variable!(T, dim, HostStorage) hy;

import mir.ndslice;;

static if(isMax)

static auto pool(S s);

alias CpuImpl = PoolRefImpl!pool;

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

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

version(grain_cuda)

Variable!(T, dim, DeviceStorage) dx;

version(grain_cuda)

Variable!(T, dim, DeviceStorage) dy;

import grain.cudnn;;

version(grain_cuda)

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

version(grain_cuda)

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

}

Members

Aliases

CpuImpl alias CpuImpl = PoolRefImpl!pool: Undocumented in source.

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.

Manifest constants

dim enum dim;: Undocumented in source.

Static functions

pool auto pool(S s): Undocumented in source. Be warned that the author may not have intended to support it.

Variables

dx Variable!(T, dim, DeviceStorage) dx;
dy Variable!(T, dim, DeviceStorage) dy;: Undocumented in source.
hx Variable!(T, dim, HostStorage) hx;
hy Variable!(T, dim, HostStorage) hy;
pad int[poolDims] pad;
stride int[poolDims] stride;: Undocumented in source.
window int[poolDims] window;: Undocumented in source.

Meta

Source

See Implementation

grain functions unary aliases

functions

manifest constants

CUDNN_ACTIVATION_IMPL_MIXIN

structs