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

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

#include <array_ops.h>

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

सारांश

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

    # Scalar indices
    output[:, ..., :] = params[indices, :, ... :]

    # Vector indices
    output[i, :, ..., :] = params[indices[i], :, ... :]

    # Higher rank indices
    output[i, ..., j, :, ... :] = params[indices[i, ..., j], :, ..., :]

अगर indices एक क्रमचय और है len(indices) == params.shape[0] तो इस कार्रवाई को दूसरे स्थान पर रखना होगा params तदनुसार।

validate_indices : निर्धारित। यदि यह ऑपरेशन सीपीयू को सौंपा गया है, तो indices में मान हमेशा सीमा के भीतर मान्य होते हैं। यदि GPU को सौंपा गया है, तो आउट-ऑफ-बाउंड सूचकांकों के परिणामस्वरूप सुरक्षित लेकिन अनिर्दिष्ट व्यवहार होता है, जिसमें त्रुटि उठाना शामिल हो सकता है।

तर्क:

रिटर्न:

  • Output : आउटपुट टेंसर।

कंस्ट्रक्टर और डिस्ट्रक्टर्स

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

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

operation
output

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

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

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

ValidateIndices (bool x)

संरचनाएं

टेंसोफ़्लो :: ऑप्स :: इकट्ठा करना: संलग्न करता है

वैकल्पिक विशेषता इकट्ठा करने के लिए

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

ऑपरेशन

Operation operation

उत्पादन

::tensorflow::Output output

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

इकट्ठा

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

इकट्ठा

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

नोड

::tensorflow::Node * node() const 

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

 operator::tensorflow::Input() const 
है

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

 operator::tensorflow::Output() const 

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

वैधता

Attrs ValidateIndices(
  bool x
)