Tôi đang cố gắng chạy suy luận CNN cho MNIST trên iOS. Có một mẫu mã tốt từ Apple để bắt đầu. https://developer.apple.com/library/content/samplecode/MPSCNNHelloWorld/Introduction/Intro.html#//apple_ref/doc/uid/TP40017482-Intro-DontLinkElementID_2Làm thế nào để tạo bộ lọc Flatten bằng cách sử dụng MPS (MetalPerformanceShaders)?
Tuy nhiên, khi tôi cố triển khai nhiều mô hình CNN phức tạp (ví dụ: https://github.com/fchollet/keras/blob/master/examples/mnist_cnn.py) bằng cách sử dụng MPS, tôi thấy không có lớp cho bộ lọc "Flatten".
Tôi đã xem qua khung MPS để tìm chức năng định hình lại hoặc thay đổi thứ nguyên nhưng tôi không thể tìm thấy thứ nguyên phù hợp. (Ví dụ MPSImageConversion dường như chỉ để chuyển đổi một màu sắc, nhưng không phải cho một chiều hướng.
Nếu ai đó biết bộ lọc cho Flatten hoặc làm thế nào để chuyển đổi hình ảnh đa chiều để 1D hình ảnh, xin vui lòng cho tôi biết.
Tôi thấy rằng lớp Flatten không cần thiết vì MPS có thể xử lý dữ liệu đa chiều làm đầu vào. – kangaroo