DJL de Java Deep Learning crea la red NiN

1. El bloque de NiN se compone de una capa convolucional y dos capas convolucionales 1 × 1.

Construya el código de bloque NiN de la siguiente manera

    public static SequentialBlock niNBlock(int numChannels, Shape kernelShape,
                                           Shape strideShape, Shape paddingShape) {

        SequentialBlock tempBlock = new SequentialBlock();
        // numChannels  通道数 滤波器层数
        //kernelShape 卷积核大小
        //strideShape 步幅
        // paddingShape  填充大小
        tempBlock.add(Conv2d.builder()
                .setKernelShape(kernelShape)
                .optStride(strideShape)
                .optPadding(paddingShape)
                .setFilters(numChannels)
                .build())
                .add(Activation::relu)
                .add(Conv2d.builder()
                        .setKernelShape(new Shape(1, 1))
                        .setFilters(numChannels)
                        .build())
                .add(Activation::relu)
                .add(Conv2d.builder()
                        .setKernelShape(new