public static Symbol Pooling(Symbol data,
Shape kernel,
PoolingPoolType poolType,
bool globalPool = false,
bool cudnnOff = false,
PoolingPoolingConvention poolingConvention = PoolingPoolingConvention.Valid)
{
return(Pooling(data, kernel, poolType, globalPool, cudnnOff, poolingConvention, new Shape()));
}
public static Symbol Pooling(Symbol data,
Shape kernel,
PoolingPoolType poolType,
bool globalPool,
bool cudnnOff,
PoolingPoolingConvention poolingConvention,
Shape stride)
{
return(Pooling(data, kernel, poolType, globalPool, cudnnOff, poolingConvention, stride, new Shape()));
}
public static Symbol Pooling(Symbol data,
Shape kernel,
PoolingPoolType poolType,
bool globalPool,
bool cudnnOff,
PoolingPoolingConvention poolingConvention,
Shape stride,
Shape pad)
{
return(new Operator("Pooling").SetParam("kernel", kernel)
.SetParam("pool_type", PoolingPoolTypeValues[(int)poolType])
.SetParam("global_pool", globalPool)
.SetParam("cudnn_off", cudnnOff)
.SetParam("pooling_convention", PoolingPoolingConventionValues[(int)poolingConvention])
.SetParam("stride", stride)
.SetParam("pad", pad)
.SetInput("data", data)
.CreateSymbol());
}