Trait dfdx::tensor_ops::GenericUpscale2D

source · 
pub trait GenericUpscale2D<M: UpscaleMethod>: HasErr {
    type Output<OH: Dim, OW: Dim>;

    // Required method
    fn generic_upscale2d_like<OH: Dim, OW: Dim>(
        self,
        method: M,
        height: OH,
        width: OW
    ) -> Result<Self::Output<OH, OW>, Self::Err>;
}

Required Associated Types§

source

type Output<OH: Dim, OW: Dim>

Required Methods§

source

fn generic_upscale2d_like<OH: Dim, OW: Dim>( self, method: M, height: OH, width: OW ) -> Result<Self::Output<OH, OW>, Self::Err>

Implementors§

source§

impl<B: Dim, C: Dim, H: Dim, W: Dim, E: Dtype, M: UpscaleMethod, D: Upscale2DKernel<E, M> + ZerosTensor<E>, T: 'static + Tape<E, D>> GenericUpscale2D<M> for Tensor<(B, C, H, W), E, D, T>

§

type Output<OH: Dim, OW: Dim> = Tensor<(B, C, OH, OW), E, D, T>

source§

impl<C: Dim, H: Dim, W: Dim, E: Dtype, M: UpscaleMethod, D: Upscale2DKernel<E, M> + ZerosTensor<E>, T: 'static + Tape<E, D>> GenericUpscale2D<M> for Tensor<(C, H, W), E, D, T>

§

type Output<OH: Dim, OW: Dim> = Tensor<(C, OH, OW), E, D, T>