Trait dfdx::tensor_ops::TryPool2D
source · pub trait TryPool2D<Kernel, Stride, Padding, Dilation>: Sized {
type Pooled;
type Error: Debug;
// Required method
fn try_pool2d(
self,
kind: Pool2DKind,
kernel: Kernel,
stride: Stride,
padding: Padding,
dilation: Dilation
) -> Result<Self::Pooled, Self::Error>;
// Provided method
fn pool2d(
self,
kind: Pool2DKind,
kernel: Kernel,
stride: Stride,
padding: Padding,
dilation: Dilation
) -> Self::Pooled { ... }
}