TF.Stack -funksjon i TensorFlow.JS

I tensorflow.JS Library, TF.Stack () -funksjon brukes til å bli med i to eller flere tensorer.

Syntaks:

Parametere:

1. Det tar to eller flere tensorer som en parameter som kan være endimensjonal som den første parameteren.
2. Aksen tar to mulige verdier (0 og 1).

Hvis spesifisert 0, TF.Stack () blir med Tensors etter hverandre ved å returnere en ny tensor.

Hvis spesifisert 1, TF.stack () blir med element etter element på rad ved å returnere en ny tensor.

Eksempel 1

Lag to 1D-tensorer med heltall og stabler to tensorer ved å stille inn Axis-0.

Produksjon:

Tensor1 og tensor2 er stablet etter hverandre.

Eksempel 2

Lag to 1D-tensorer med heltall og stabler to tensorer ved å stille inn Axis-1.

Produksjon:

Tensor1 og Tensor2 er stablet element etter element i begge tensorer, og returnert Tensor har stablet elementer plassert i separate rader.

Eksempel 3

Lag fire 1D-tensorer med heltall og stabler dem ved å stille inn Axis-1 og Axis-0 hver for seg.

Produksjon:

I den første utgangen er fire tensorer stablet element etter element, og i den andre utgangen er tensorer stablet om gangen.

Konklusjon

I denne tensorflow.JS -opplæring, vi så hvordan du kan stable to eller flere tensorer ved hjelp av TF.Stack () Funksjon. Hvis aksen er spesifisert som 0, TF.Stack () blir med Tensors etter hverandre ved å returnere en ny tensor. Hvis spesifisert som 1, TF.stack () blir med element etter element på rad ved å returnere en ny tensor. Forsikre deg om at du forstår alle eksemplene som er diskutert og får forskjellen mellom Axis-0/1-parametere.