मैं दूसरी अक्ष के साथ 25 x 5 x 5 मैट्रिक्स (टेंसर) का तर्क कैसे दे सकता हूं? अनिवार्य रूप से, मैं tensorflow के बराबर (फ़ंक्शन या पद्धति) को numpy के argsort में देख रहा हूं, उदा। np.argsort(matrix, 2)।टेन्सफोर्लो में Argsort कैसे करें?
अपने मामले में आप शायद top_k का उपयोग कर सकते हैं जो उच्चतम k मान देता है। k 1 डी वेक्टर परिभाषित कर सकता है कि प्रति आयाम 'शीर्ष' के लिए कितने मान हैं। आपके मामले में, यदि आप चाहते हैं कि दूसरा अक्ष सेट k=[0, 5, 0] ऐसा कर सकता है।
tf.nn.top_k(matrix, k=[0,5,0], sorted=True)
मैंने इसे नहीं चलाया। उम्मीद है कि यह
में मदद करता है