יום הקהילה ML הוא 9 בנובמבר! הצטרפו אלינו עדכונים מ- TensorFlow, JAX, ועוד למידע נוסף

זרימת טנסור :: אופ :: לאסוף

#include <array_ops.h>

אסוף פרוסות params לפי indices .

סיכום

indices חייבים להיות טנזור שלם של כל מימד (בדרך כלל 0-D או 1-D). מייצר טנסור פלט עם indices.shape + params.shape[1:] צורה indices.shape + params.shape[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] אז הפעולה הזו תהיה permute params בהתאם.

validate_indices : DEPRECATED. אם פעולה זו מוקצית ל- CPU, הערכים indices מאומתים תמיד להיות בטווח. אם הם מוקצים ל- GPU, מדדים מחוץ לתחום גורמים להתנהגות בטוחה אך לא מוגדרת, אשר עשויה לכלול העלאת שגיאה.

טענות:

  • היקף: אובייקט Scope

החזרות:

  • 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)

סטרוקטורים

tensorflow :: ops :: התכנסו :: Attrs

קובעי תכונות אופציונליים עבור Gather .

תכונות ציבוריות

מבצע

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 

פונקציות סטטיות ציבוריות

ValidateIndices

Attrs ValidateIndices(
  bool x
)