1、伪代码
Model model = Model.newInstance(MODEL_NAME);
// Block is a composable unit that forms a neural network; combine them like Lego blocks
// to form a complex network
Block block = new SequentialBlock()
.add(Conv2d.builder()
//卷积核
.setKernelShape(new Shape(5, 5))
//步幅
.optStride(new Shape(5, 5))
//padding
.optPadding(new Shape(2, 2))
//过滤器
.setFilters(3)
.build())
.add(Activation::relu)
.add(Conv2d.builder()
.setKernelShape(new Shape(5, 5))
.optStride(new Shape(3, 3))
.optPadding(new Shape(1, 1))
.setFilters(3)
.build())