main function

void main()

Implementation

void main() {
  // Example feature maps (3-channel, each 3x3)
  List<Matrix2d> featureMaps = [
    Matrix2d(3, 3), // Feature map 1
    Matrix2d(3, 3), // Feature map 2
    Matrix2d(3, 3), // Feature map 3
  ];

  // Create Flatten layer
  Flatten flatten = Flatten();

  // Apply flattening
  ValueVector flattenedOutput = flatten.forward(featureMaps);

  print("Flattened Output Size: ${flattenedOutput.values.length}");
}