Was ist die Alternative von numpy.newaxis im Tensorflow?

Ich denke, das wäre tf.expand_dims -

Im Grunde geben wir die Achsen-ID an, wo diese neue Achse eingefügt werden soll, und die nachfolgenden Achsen / Dims werden zurückgeschoben .

In den verknüpften Dokumenten finden Sie hier einige Beispiele für expandierende Dimensionen -

Der entsprechende Befehl lautet tf.newaxis . Es hat keinen eigenen Eintrag in der Tensorflow-Dokumentation, wird aber kurz auf der Doc-Seite von tf.stride_slice .

Die Verwendung von tf.expand_dims ist ebenfalls in Ordnung, aber wie im obigen Link angegeben,

  

Diese Schnittstellen sind viel freundlicher und sehr empfehlenswert.

Wenn Sie an genau demselben Typ interessiert sind (d. h. None ) wie in NumPy, dann ist tf.newaxis die exakte Alternative zu np.newaxis .

Beispiel:

Dies ist genau die gleiche Art von Operationen, die Sie in NumPy durchführen. Verwenden Sie es genau in der gleichen Dimension, in der es vergrößert werden soll.