Angenommen x
, enthält Segment-IDs, ich möchte jedem Element innerhalb jeder Segment-ID eine eindeutige ID geben. Dies muss in einer tensorflow
Operation ausgeführt werden
x = tf.constant([1, 1, 2, 2, 3, 3, 4, 1])
Benötigte Leistung:
[0, 1, 0, 1, 0, 1, 0, 2]
Zählt einfach jeden Artikel innerhalb jeder Segment-ID. Ich möchte keine verwenden py_func
.
point_ids = tf.zeros_like(x)
for i in range(n_pillars):
indicies = tf.cast(tf.where(tf.equal(x, i)), tf.int32)
updates = tf.range(len(indicies))
point_ids += tf.scatter_nd(indicies, updates, shape=tf.shape(idx))
Dieser Artikel stammt aus dem Internet. Bitte geben Sie beim Nachdruck die Quelle an.
Bei Verstößen wenden Sie sich bitte [email protected] Löschen.
Lass mich ein paar Worte sagen