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

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

#include <array_ops.h>

מכניס מימד של 1 לצורת טנסור.

סיכום

בהינתן input טנסור, פעולה זו מכניסה מימד של 1 axis אינדקס הממד של צורת input . axis אינדקס המימדים מתחיל מאפס; אם אתה מציין מספר שלילי axis זה נספר אחורה מהסוף.

פעולה זו שימושית אם ברצונך להוסיף מימד אצווה לאלמנט יחיד. לדוגמא, אם יש לך תמונה אחת של צורה [height, width, channels] , תוכל להפוך אותה expand_dims(image, 0) של תמונה אחת עם expand_dims(image, 0) , מה שיהפוך את הצורה [1, height, width, channels] .

דוגמאות נוספות:

# 't' is a tensor of shape [2]
shape(expand_dims(t, 0)) ==> [1, 2]
shape(expand_dims(t, 1)) ==> [2, 1]
shape(expand_dims(t, -1)) ==> [2, 1]

# 't2' is a tensor of shape [2, 3, 5]
shape(expand_dims(t2, 0)) ==> [1, 2, 3, 5]
shape(expand_dims(t2, 2)) ==> [2, 3, 1, 5]
shape(expand_dims(t2, 3)) ==> [2, 3, 5, 1]

פעולה זו מחייבת:

-1-input.dims() <= dim <= input.dims()

פעולה זו קשורה squeeze() , המסירה ממדים בגודל 1.

טענות:

  • היקף: אובייקט Scope
  • ציר: 0-D (סקלרי). מציין את אינדקס המימדים בו ניתן להרחיב את צורת input . חייב להיות בטווח [-rank(input) - 1, rank(input)] .

החזרות:

  • Output : מכיל את אותם נתונים כמו input , אך לצורתו נוסף ממד נוסף בגודל 1.

קונסטרוקטורים ומשחתנים

ExpandDims (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input axis)

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

operation
output

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

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

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

מבצע

Operation operation

תְפוּקָה

::tensorflow::Output output

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

ExpandDims

 ExpandDims(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input axis
)

צוֹמֶת

::tensorflow::Node * node() const 

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

 operator::tensorflow::Input() const 

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

 operator::tensorflow::Output() const