tensorflow:: ops:: SparseSplit

#include <sparse_ops.h>

פיצול SparseTensor לטנזורים מסוג num_split לאורך ממד אחד.

תַקצִיר

אם shape[split_dim] אינו כפולה שלמה של num_split . פרוסות [0 : shape[split_dim] % num_split] מקבלות ממד נוסף אחד. לדוגמה, אם split_dim = 1 ו- num_split = 2 והקלט הוא

input_tensor = shape = [2, 7]
[    a   d e  ]
[b c          ]

מבחינה גרפית, טנזורי הפלט הם:

output_tensor[0] = shape = [2, 4]
[    a  ]
[b c    ]

output_tensor[1] = shape = [2, 3]
[ d e  ]
[      ]

טיעונים:

  • היקף: אובייקט היקף
  • split_dim: 0-D. הממד שלאורכו יש לפצל. חייב להיות בטווח [0, rank(shape)) .
  • אינדקסים: טנזור דו-ממדי מייצג את האינדקסים של הטנזור הדליל.
  • ערכים: טנזור חד-ממדי מייצג את ערכי הטנזור הדליל.
  • צורה: טנזור חד-ממדי מייצג את צורת הטנזור הדליל. אינדקסי פלט: רשימה של טנזורים חד-ממדיים מייצגים את האינדקסים של טנזורי הדלילים בפלט.
  • num_split: מספר הדרכים לפיצול.

החזרות:

  • OutputList מדדי פלט
  • רשימת ערכי_פלט OutputList : רשימה של טנזורים חד-ממדיים מייצגים את הערכים של טנזורים דלילים בפלט.
  • OutputList output_shape: רשימה של טנזורים חד-ממדיים מייצגים את צורת הטנזורים הדלילים של הפלט.

בנאים והורסים

SparseSplit (const :: tensorflow::Scope & scope, :: tensorflow::Input split_dim, :: tensorflow::Input indices, :: tensorflow::Input values, :: tensorflow::Input shape, int64 num_split)

מאפיינים ציבוריים

operation
output_indices
output_shape
output_values

מאפיינים ציבוריים

מִבצָע

Operation operation

מדדי_פלט

::tensorflow::OutputList output_indices

צורת_פלט

::tensorflow::OutputList output_shape

ערכי_פלט

::tensorflow::OutputList output_values

תפקידים ציבוריים

SparseSplit

 SparseSplit(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input split_dim,
  ::tensorflow::Input indices,
  ::tensorflow::Input values,
  ::tensorflow::Input shape,
  int64 num_split
)