Tensorflow.js tf.atan2()用法及代碼示例

Tensorflow.js是一個開放源代碼庫，由Google開發，用於在瀏覽器或節點環境中運行機器學習模型以及深度學習神經網絡。

.atan2()函數用於查找指定張量輸入的反正切值，並逐元素進行。此外，它還有助於廣播。

用法：

tf.atan2(a, b)

Parameters:

a:它是第一個張量輸入，其類型可以為tf.Tensor，TypedArray或Array。
b:它是第二張量輸入，其類型可以為tf.Tensor，TypedArray或Array，並且必須具有與a相同的數據類型。

返回值：它返回tf.Tensor對象。

範例1：在此示例中，我們定義了整數類型的輸入張量，然後打印其atan2值。為了創建輸入張量，我們使用.tensor1d()方法，為了打印輸出，我們使用.print()方法。

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining first tensor input elements
const y = tf.tensor1d([9, 10, 11, 12]);
  
// Defining second tensor input elements
const z = tf.tensor1d([13, 14, 15, 17]);
  
// Calling atan2() method and
// Printing output
(y).atan2(z).print();

輸出：

Tensor
    [0.6055371, 0.6202452, 0.6327478, 0.6146573]

範例2：在此示例中，浮點類型值被視為張量輸入，並且兩個參數都直接傳遞給atan2函數。

Javascript


// Importing the tensorflow.js library 
import * as tf from "@tensorflow/tfjs"
  
// Defining float values for first tensor
var val1 = [.5, 4.9, .275];
  
// Defining float values for second tensor
var val2 = [1.5, 5.9, .775];
  
// Calling tensor1d method
const y = tf.tensor1d(val1);
const z = tf.tensor1d(val2);
  
// Calling atan2() method
var res = tf.atan2(y,z);
  
// Printing output
res.print();

輸出：

Tensor
    [0.3217588, 0.6930802, 0.340989]

參考：https://js.tensorflow.org/api/latest/#atan2

相關用法

注：本文由純淨天空篩選整理自nidhi1352singh大神的英文原創作品 Tensorflow.js tf.atan2() Function。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。