टेंसरफ़्लो:: ऑप्स:: इकट्ठा V2

#include <array_ops.h>

indices के अनुसार params अक्ष axis से स्लाइस इकट्ठा करें।

सारांश

indices किसी भी आयाम (आमतौर पर 0-डी या 1-डी) का एक पूर्णांक टेंसर होना चाहिए। आकार के साथ एक आउटपुट टेंसर का उत्पादन करता है params.shape[:axis] + indices.shape + params.shape[axis + 1:] जहां:

    # Scalar indices (output is rank(params) - 1).
    output[a_0, ..., a_n, b_0, ..., b_n] =
      params[a_0, ..., a_n, indices, b_0, ..., b_n]

    # Vector indices (output is rank(params)).
    output[a_0, ..., a_n, i, b_0, ..., b_n] =
      params[a_0, ..., a_n, indices[i], b_0, ..., b_n]

    # Higher rank indices (output is rank(params) + rank(indices) - 1).
    output[a_0, ..., a_n, i, ..., j, b_0, ... b_n] =
      params[a_0, ..., a_n, indices[i, ..., j], b_0, ..., b_n]

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

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

तर्क:

• स्कोप: एक स्कोप ऑब्जेक्ट
• पैराम्स: वह टेन्सर जिससे मूल्यों को इकट्ठा किया जाता है। कम से कम रैंक axis + 1 होना चाहिए।
• सूचकांक: सूचकांक टेंसर। [0, params.shape[axis]) की सीमा में होना चाहिए।
• अक्ष: indices को इकट्ठा करने के लिए params में अक्ष। पहले आयाम के लिए डिफ़ॉल्ट। नकारात्मक सूचकांक का समर्थन करता है।

रिटर्न:

• Output : आकार के साथ indices द्वारा दिए गए सूचकांकों से एकत्र किए गए पैरा से मान params params.shape[:axis] + indices.shape + params.shape[axis + 1:] ।

निर्माता और विनाशक
GatherV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input params, :: tensorflow::Input indices, :: tensorflow::Input axis)
GatherV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input params, :: tensorflow::Input indices, :: tensorflow::Input axis, const GatherV2::Attrs & attrs)

सार्वजनिक गुण
operation	Operation
output	:: tensorflow::Output

सार्वजनिक समारोह
node () const	::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

सार्वजनिक स्थैतिक कार्य
BatchDims (int64 x)	Attrs

स्ट्रक्चर्स
tensorflow:: ops:: GatherV2:: Attrs	GatherV2 के लिए वैकल्पिक विशेषता बसती है।

सार्वजनिक गुण

संचालन

Operation operation

उत्पादन

::tensorflow::Output output

सार्वजनिक समारोह

इकट्ठा V2

 GatherV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input params,
  ::tensorflow::Input indices,
  ::tensorflow::Input axis
)

इकट्ठा V2

 GatherV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input params,
  ::tensorflow::Input indices,
  ::tensorflow::Input axis,
  const GatherV2::Attrs & attrs
)

नोड

::tensorflow::Node * node() const 

ऑपरेटर :: टेंसरफ़्लो :: इनपुट

 operator::tensorflow::Input() const 

ऑपरेटर :: टेंसरफ़्लो :: आउटपुट

 operator::tensorflow::Output() const 

सार्वजनिक स्थैतिक कार्य

बैचडिम्स

Attrs BatchDims(
  int64 x
)