Wie lösche ich Einträge in einem Tensor in TensorFlow?

Question

Nach einigen Berechnungen erhalte ich einen Tensor x der Form (128, 128, 32). Wie kann ich das gleiche Verhalten wie im nächsten Code erreichen?

x[:, :, 1:] = 0

Praktisch würde Ich mag alle Einträge löschen, außer für eine Matrix.

Der Tensor wird innerhalb einer Sitzung erhalten und ich möchte die Einträge in derselben Sitzung löschen.

Was ist der einfachste Weg, dies zu tun?

Answer 1

Ab 0.10 verfügt TensorFlow über keine erweiterte Indizierung, die für diesen Zweck verwendet werden kann, daher müssen Sie den Vektor umformen, der Menge der linearen Indizes zuweisen und dann die Form neu gestalten. Sie können diese Zuweisung für einen Vektor ausführen, indem Sie die Variablen + scatter_update oder dynamic_stitch wie angegeben verwenden here