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