class MXNet::Gluon::NN::Internal::Pooling

Overview

Base class for pooling layers.

Direct Known Subclasses

Defined in:

mxnet/gluon/nn/layers.cr

Constructors

Instance Method Summary

Instance methods inherited from class MXNet::Gluon::HybridBlock

export(filename, epoch = 0) export, forward(inputs : Array(T)) : Array(T) forall T forward, hybrid_forward(inputs : Array(T), params : Hash(String, T) = {} of String => T) : Array(T) forall T hybrid_forward, hybridize(active = true, flags = {} of String => String) hybridize, register_child(block, name = nil) register_child

Instance methods inherited from module MXNet::Gluon::CachedGraph

clear_cache clear_cache, infer_dtype(args) infer_dtype, infer_shape(args) infer_shape

Constructor methods inherited from module MXNet::Gluon::CachedGraph

new(**kwargs) new

Instance methods inherited from class MXNet::Gluon::Block

call(inputs : Array(T)) : Array(T) forall T call, children children, collect_params(selector = nil) collect_params, forward(inputs : Array(T)) : Array(T) forall T forward, get_attr(name : String) : Block | Parameter | Nil get_attr, hybridize(active = true) hybridize, init(init = nil, ctx = nil, force_reinit = false) init, load_parameters(fname, ctx = MXNet.cpu, allow_missing = false, ignore_extra = false) load_parameters, params : MXNet::Gluon::ParameterDict params, prefix : String prefix, register_child(block, name = nil) register_child, register_parameter(param, name = nil) register_parameter, save_parameters(fname) save_parameters, scope : MXNet::Gluon::BlockScope? scope, set_attr(name : String, value : Block | Parameter | Nil) set_attr, with_name_scope(&) with_name_scope

Constructor methods inherited from class MXNet::Gluon::Block

new(prefix = nil, params = nil) new

Constructor Detail

def self.new(*, pool_size : Array(Int32), strides : Array(Int32) | Int32 | Nil, padding : Array(Int32) | Int32, **kwargs) #

Creates a new instance.

N is the number of dimensions of the pooling layer.

Parameters

  • pool_size (Array(Int32) of N integers) Specifies the dimensions of pooling operation.
  • strides (Int32 or Array(Int32) of N integers) Specifies the strides of the pooling operation.
  • padding (Int32 or Array(Int32) of N integers) If padding is non-zero, then the input is implicitly zero-padded on both sides for padding number of points.

[View source]

Instance Method Detail

def hybrid_forward(inputs : Array(T), params : Hash(String, T)? = nil) : Array(T) forall T #

[View source]