main function

void main()

Implementation

void main() {
  Matrix2d input = Matrix2d(5, 5); // Example 5x5 input matrix
  Conv2d convLayer = Conv2d(
      inChannels: 1, outChannels: 2, kernelSize: 3, stride: 1, padding: 1);

  Matrix2d output = convLayer.forward(input);
  print("Output Matrix: ${output.data!.values}");
}