Der tf.Stapelfunktion im Tensorflow.JS

Im Tensorflow.JS Library, die TF.Stack () -Funktion wird verwendet, um zwei oder mehr Tensoren zu verbinden.

Syntax:

Parameter:

1. Es dauert zwei oder mehr Tensoren als Parameter, der als erster Parameter eindimensional sein kann.
2. Die Achse erfordert zwei mögliche Werte (0 und 1).

Wenn angegeben 0, tf.Stack () schließt sich den Tensors nacheinander an, indem er einen neuen Tensor zurückgibt.

Wenn angegeben 1, tf.Stack () verbindet das Element nach Element in einer Zeile, indem er einen neuen Tensor zurückgibt.

Beispiel 1

Erstellen Sie zwei 1D-Tensoren mit Ganzzahlen und stapeln Sie zwei Tensoren, indem Sie Achse-0 einstellen.

Ausgang:

Tensor1 und Tensor2 sind nacheinander gestapelt.

Beispiel 2

Erstellen Sie zwei 1D-Tensoren mit Ganzzahlen und stapeln Sie zwei Tensoren, indem Sie Achse-1 einstellen.

Ausgang:

Tensor1 und Tensor2 sind in beiden Tensoren ein Element gestapelt, und der zurückgegebene Tensor hat Elemente gestapelt, die in getrennten Zeilen platziert sind.

Beispiel 3

Erstellen Sie vier 1D-Tensoren mit Ganzzahlen und stapeln Sie sie, indem Sie Achse-1 und Axis-0 separat einstellen.

Ausgang:

Im ersten Ausgang sind vier Tensoren von Element gestapelt, und im zweiten Ausgang werden Tensoren gleichzeitig gestapelt.

Abschluss

In diesem Tensorflow.JS Tutorial sahen wir, wie man zwei oder mehr Tensoren mit dem TF stapelt.Stack () Funktion. Wenn die Achse als 0 angegeben ist, tf.Stack () schließt sich den Tensors nacheinander an, indem er einen neuen Tensor zurückgibt. Wenn als 1 angegeben, tf.Stack () verbindet das Element nach Element in einer Zeile, indem er einen neuen Tensor zurückgibt. Stellen Sie sicher.