tf.suma()

tf.sum () w Tensorflow.JS zwraca całkowitą sumę elementów.

Składnia:

tf.sum (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ć sumę przez rzędy lub kolumny.

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

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

Powrót:

Zwróć tensor z całkowitą sumą.

Przykład 1:

Utwórzmy jednowymiarowy tensor w JS, który ma wartości liczb całkowitych i zwróć całkowitą sumę.

Wskazówka Linux

Tensorflow.JS - TF.suma()

Wyjście:

Pracujący:

34+56+78+90 = 258.

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ą sumę w kolumnach.

Wskazówka Linux

Tensorflow.JS - TF.suma()

Wyjście:

Pracujący:

Tensor [[34, 56], [78, 90], [1, 0], [3, 4]]
=>
34+78+1+0 = 116
56+90+0+4 = 150.

Przykład 3:

Utwórzmy tensor, który ma 2 wymiary w JS z 2 wierszami i 2 kolumnami, które mają wartości liczb całkowitych i zwracają sumę na rzędach.

Wskazówka Linux

Tensorflow.JS - TF.suma()

Wyjście:

Pracujący:

Tensor [[1, 0], [3, 4]]
=>
1+0 = 1
3+4 = 7.

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ą sumę we wszystkich wierszach i kolumnach.

Wskazówka Linux

Tensorflow.JS - TF.suma()

Wyjście:

Pracujący:

Tensor [[34, 56], [78, 90], [1, 0], [3, 4]]
=>
34+56+78+90+1+0+3+4 = 266.

Wniosek

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