TF.Funkcja stosu w TensorFlow.JS

W tensorflow.biblioteka JS, TF.Funkcja Stack () służy do łączenia dwóch lub więcej tensorów.

Składnia:

Parametry:

1. Zajmuje dwa lub więcej tensorów jako parametr, który może być jednowymiarowy jako pierwszy parametr.
2. Oś wymaga dwóch możliwych wartości (0 i 1).

Jeśli określono 0, TF.Stack () dołącza do tensorów jeden po drugim, zwracając nowy tensor.

Jeśli określono 1, TF.Stack () łączy element przez element z rzędu, zwracając nowy tensor.

Przykład 1

Utwórz dwa tensory 1D z liczbami całkowitymi i układaj dwa tensory, ustawiając oś.

Wyjście:

Tensor1 i Tensor2 są ułożone jeden po drugim.

Przykład 2

Utwórz dwa tensory 1D z liczbami całkowitymi i układaj dwa tensory, ustawiając oś 1.

Wyjście:

Tensor1 i tensor2 są ułożone w ułożone element w obu tensorach, a zwrócony tensor ułożył elementy umieszczone w oddzielnych rzędach.

Przykład 3

Utwórz cztery tensory 1D z liczbami całkowitymi i ułóż je, ustawiając osi-1 i osi-0 osobno.

Wyjście:

Na pierwszym wyjściu cztery tensory są ułożone w stosy według elementu, aw drugim wyjściu tensory są układane jednocześnie.

Wniosek

W tym tensorflow.Samouczek JS, widzieliśmy, jak układać dwa lub więcej tensorów za pomocą TF.Funkcja stosu (). Jeśli oś jest określona jako 0, tf.Stack () dołącza do tensorów jeden po drugim, zwracając nowy tensor. Jeśli określono jako 1, TF.Stack () łączy element przez element z rzędu, zwracając nowy tensor. Upewnij się, że rozumiesz wszystkie omówione przykłady i uzyskaj różnicę między parametrami osi-0/1.