Składnia:

tf.średnia (tensor_input, oś)

Parametr:

1. tensor_input to tensor, który ma elementy numeryczne.

Może być 1 lub 2 wymiarowy.

2. Jeśli tensor jest dwuwymiarowy, to możliwe jest określenie osi, aby uzyskać średnią w ramach rzędów lub kolumn.

Jeśli oś = 0, całkowita średnia jest zwracana pod względem kolumny, a jeśli oś = 1, całkowita średnia jest zwracana.

Jeśli oś nie jest określona, ​​zwróci średnią wszystkich elementów.

Powrót

Zwróć tensor ze średnią.

Przykład 1:

Utwórzmy jednowymiarowy tensor w JS, który ma wartości liczb całkowitych i średnią zwrotną.

Wskazówka Linux

Tensorflow.JS - TF.mieć na myśli()

Wyjście:

Pracujący:

(34+56+78+90)/4 = 64.5

Przykład 2:

Utwórzmy tensor, który ma 2 wymiary w JS z 4 wierszami i 2 kolumnami, które mają wartości liczb całkowitych i zwracają średnią w kolumnach.

Wskazówka Linux

Tensorflow.JS - TF.mieć na myśli()

Wyjście:

Pracujący:

Tensor [[1,2], [3,4], [5,6], [7,8]]
=>
(1+3+5+7)/4 => 16/4 = 4
(2+4+6+8)/4 => 20/4 = 5

Przykład 3:

Utwórzmy tensor, który ma 2 wymiary w JS z 1 wierszem i 2 kolumnami, które mają wartości liczb całkowitych i zwracają średnią w wierszach.

Wskazówka Linux

Tensorflow.JS - TF.mieć na myśli()

Wyjście:

Pracujący:

Tensor [[1], [3]]
=>
1
3.

Ponieważ w każdym rzędzie jest tylko jeden element, sam powraca.

Przykład 4:

Utwórzmy tensor, który ma 2 wymiary w JS z 4 wierszami i 2 kolumnami, które mają wartości liczb całkowitych i zwracają całkowitą średnią we wszystkich wierszach i kolumnach.

Wskazówka Linux

Tensorflow.JS - TF.mieć na myśli()

Wyjście:

Pracujący:

Tensor [[34, 56], [78, 90], [1, 0], [3, 4]]
=>
(34+56+78+90+1+0+3+4)/8 = 33.25.

Wniosek

W tym tensorflow.Samouczek JS, widzieliśmy, jak zwrócić całkowitą średnią elementów obecnych w tensorze za pomocą TF.Metoda średniej (). W tensorze 2D, jeśli oś = 0, całkowita średnia jest zwracana pod względem kolumny, a jeśli oś = 1, całkowita średnia jest zwracana. Domyślnie zwróci średnią wszystkich elementów w rzędach i kolumnach.