clusterکردن دادهها درون یک تنسور ۴بعدی؟

Question

سلام فرض کنید یک تنسور ۴ بعدی مثل تنسور زیر داریم:

image1 = tf.constant([ [[[10, 11,3],[1,13,10],[11,3,13],[1,23,1]],

               [[10, 11,3],[1,13,14],[32,23,15],[1,23,1]],
               [[10, 11,3],[1,1۰3,10],[11,3,13],[1,23,1]],
               [[14, 32, 23],[10, 11,3],[1,13,3],[1,23,۱‍‍۱1]],
               [[10, 11, 3], [1, 13, 10],[11, 3, 1],[1,23,1]],
               [[10, 11, 3], [1, ۱۲۴, 14],[32, 23, ۴3],[1,23,1]],
                  ],
              [ [[10, 11,3۸],[1,13,10],[11,3,13],[1,23,1]],
               [[10, 11,۶3],[1,13,10],[11,3,13],[1,23,1]],
               [[10, 11,3],[1,13,1۵۰],[32,23,18],[1,23,1]],
               [[14, 32, 23],[10, 11,3],[1,1۶3,3],[1,23,1]],
               [[10, 11, 3], [1, 13۰, 10],[11۵, 3۷, 1],[1,23۰,1]],
               [[10, 11, 3], [200, 13, 14],[32, 23۰, 3],[1,23,1]],
                ]]
                 )

حالا می خواهیم اعداد درون ین تنسور را دسته بندی کنیم. مثلا می خواهیم به ۴ قسمت تقسیمشون کنیم ( اعداد بین ۰ تا ۵۰ در یک دسته ُ‌اعداد بین ۵۰ تا ۱۰۰ و اعداد بین ۱۰۰ تا ۱۵۰ و اعداد بین ۱۵۰ تا 200 در دسته های مجزا قرار گیرند و به صورت ماتریسی با همین ابعاد برای ما خروجی داشته باشه . چه باید کرد ؟