Google I/O एक लपेट है! TensorFlow सत्रों पर पकड़ बनाएं सत्र देखें

टेंसरफ़्लो :: ऑप्स :: GatherV2

#include <array_ops.h>

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

सारांश

indices किसी भी आयाम (आमतौर पर 0-डी या 1-डी) का पूर्णांक दशांश होना चाहिए। आकार params.shape[:axis] + indices.shape + params.shape[axis + 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]

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

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

तर्क:

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

रिटर्न:

  • Output : से मान params द्वारा दिए गए सूचकांक से इकट्ठा indices , आकार के साथ 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
output

सार्वजनिक कार्य

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

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

BatchDims (int64 x)

संरचनाएं

टेंसोफ़्लो :: ऑप्स :: GatherV2 :: एट्र्स

वैकल्पिक विशेषता GatherV2 के लिए बसती है

सार्वजनिक विशेषताएँ

ऑपरेशन

Operation operation

उत्पादन

::tensorflow::Output output

सार्वजनिक कार्य

GatherV2

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

GatherV2

 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
)