use dfdx::prelude::*;

const IN_CHAN: usize = 1;
const OUT_CHAN: usize = 1;
const KERNEL_SIZE: usize = 3;
const STRIDE: usize = 2;
// for padding='same'
const PADDING: usize = ((KERNEL_SIZE - 1) * DILATION + 1) / 2; // = 1
const DILATION: usize = 1;
const GROUPS: usize = 1;
type Model = ConvTrans2DConstConfig<
    IN_CHAN,
    OUT_CHAN,
    KERNEL_SIZE,
    STRIDE,
    PADDING,
    DILATION,
    GROUPS,
>;

fn example() {
    let dev = Cpu::default();
    let model = dev.build_module::<f32>(Model::default());
    let x: Tensor<Rank4<1, 1, 2, 3>, _, _> = dev.zeros();
    let _prediction: Tensor<Rank4<1, 1, 3, 5>, _, _> = model.forward(x);
    // note: the shape is the same for `ya` from the tf example.
}