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

Tensorflow.js是Google開發的一個開放源代碼庫，用於在瀏覽器或節點環境中運行機器學習模型和深度學習神經網絡。。它還可以幫助開發人員使用JavaScript語言開發ML模型，並且可以直接在瀏覽器或Node.js中使用ML。

tf.slice()函數用於從tf.Tensor中提取一個切片，該切片起始於坐標‘begin’，大小為‘size’。

用法:

tf.slice (x, begin, size?)

參數：

• x:輸入張量到切片形式。
• begin:從其開始切片的坐標。
• size:切片的大小

返回值：它返回tf.Tensor。

範例1：

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
const x = tf.tensor1d([1, 2, 3, 4, 5, 6, 7]);
  
x.slice([0], [4]).print();

輸出：

Tensor
    [1, 2, 3, 4]

範例2：

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
const x = tf.tensor2d([1, 2, 3, 4, 5, 6], [2, 3]);
  
x.slice([1, 0], [1, 2]).print();

輸出：

Tensor
     [1,2,3,4]

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