GatherNd

`पैराम्स` से स्लाइस को `इंडिसेज़` द्वारा निर्दिष्ट आकार के साथ एक टेंसर में इकट्ठा करें।

`सूचकांक` एक K-आयामी पूर्णांक टेंसर है, जिसे `params` में सूचकांकों के (K-1)-आयामी टेंसर के रूप में सबसे अच्छा माना जाता है, जहां प्रत्येक तत्व `params` के एक स्लाइस को परिभाषित करता है:

आउटपुट[\\(i_0, ..., i_{K-2}\\)] = पैरामीटर्स[सूचकांक[\\(i_0, ..., i_{K-2}\\)]]

जबकि tf.gather में `indices` स्लाइस को `params` के `axis` आयाम में परिभाषित करता है, tf.gather_nd में, `indices` स्लाइस को `params` के पहले `N` आयामों में परिभाषित करता है, जहां `N = indices.shape [-1]`.

`सूचकांकों` का अंतिम आयाम अधिकतम `पैराम` की रैंक हो सकता है:

सूचकांक.आकार[-1] <= पैरामीटर.रैंक

`सूचकांक` का अंतिम आयाम तत्वों से मेल खाता है (यदि `indices.shape[-1] ==params.rank`) या स्लाइस (यदि `indices.shape[-1] <params.rank`) आयाम `सूचकांकों के साथ। `params` का आकार[-1]`। आउटपुट टेंसर का आकार होता है

सूचकांक आकार[:-1] + पैरामीटर आकार[सूचकांक आकार[-1]:]

ध्यान दें कि सीपीयू पर, यदि कोई आउट ऑफ बाउंड इंडेक्स पाया जाता है, तो एक त्रुटि वापस आ जाती है। जीपीयू पर, यदि आउट ऑफ बाउंड इंडेक्स पाया जाता है, तो संबंधित आउटपुट मान में 0 संग्रहीत किया जाता है।

नीचे कुछ उदाहरण.

मैट्रिक्स में सरल अनुक्रमण:

indices = [[0, 0], [1, 1]]
     params = [['a', 'b'], ['c', 'd']]
     output = ['a', 'd']
 

मैट्रिक्स में स्लाइस अनुक्रमण:

indices = [[1], [0]]
     params = [['a', 'b'], ['c', 'd']]
     output = [['c', 'd'], ['a', 'b']]
 

indices = [[1]]
     params = [[['a0', 'b0'], ['c0', 'd0']],
               [['a1', 'b1'], ['c1', 'd1']]]
     output = [[['a1', 'b1'], ['c1', 'd1']]]
 
 
     indices = [[0, 1], [1, 0]]
     params = [[['a0', 'b0'], ['c0', 'd0']],
               [['a1', 'b1'], ['c1', 'd1']]]
     output = [['c0', 'd0'], ['a1', 'b1']]
 
 
     indices = [[0, 0, 1], [1, 0, 1]]
     params = [[['a0', 'b0'], ['c0', 'd0']],
               [['a1', 'b1'], ['c1', 'd1']]]
     output = ['b0', 'b1']
 

3-टेंसर में अनुक्रमण:

indices = [[[0, 0]], [[0, 1]]]
     params = [['a', 'b'], ['c', 'd']]
     output = [['a'], ['b']]
 

मैट्रिक्स में बैच अनुक्रमण: l10n-प्लेसहोल्डर6 मैट्रिक्स में बैच अनुक्रमण:

indices = [[[1]], [[0]]]
     params = [['a', 'b'], ['c', 'd']]
     output = [[['c', 'd']], [['a', 'b']]]
 

बैच अनुक्रमण एक 3-टेंसर:

indices = [[[1]], [[0]]]
     params = [[['a0', 'b0'], ['c0', 'd0']],
               [['a1', 'b1'], ['c1', 'd1']]]
     output = [[[['a1', 'b1'], ['c1', 'd1']]],
               [[['a0', 'b0'], ['c0', 'd0']]]]
 
     indices = [[[0, 1], [1, 0]], [[0, 0], [1, 1]]]
     params = [[['a0', 'b0'], ['c0', 'd0']],
               [['a1', 'b1'], ['c1', 'd1']]]
     output = [[['c0', 'd0'], ['a1', 'b1']],
               [['a0', 'b0'], ['c1', 'd1']]]
 
 
     indices = [[[0, 0, 1], [1, 0, 1]], [[0, 1, 1], [1, 1, 0]]]
     params = [[['a0', 'b0'], ['c0', 'd0']],
               [['a1', 'b1'], ['c1', 'd1']]]
     output = [['b0', 'b1'], ['d0', 'c1']]
 

tf.gather और `tf.batch_gather` भी देखें।