Tensorflow.js tf.data.Dataset.skip()用法及代码示例

Tensorflow.js是一个开放源代码库，由Google开发，用于在浏览器或节点环境中运行机器学习模型以及深度学习神经网络。

tf.data.Dataset.skip() 方法用于创建一个数据集，该数据集从该数据集中跳过计数初始元素。

用法:

skip(count)

参数：此方法具有如上所述和如下所述的单个参数：

count:它是一个张量输入，其中应跳过此数据集的元素数以形成新数据集。当计数大于此数据集的大小时，新数据集将不包含任何元素。当计数未定义或为负时，它会跳过整个数据集。

返回值：它返回 tf.data.Dataset。

以下示例演示了 tf.data.Dataset.skip() 方法：

范例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining input elements
const a = 
  tf.data.array([4, 5, 6, 7, 8, 9]).skip(3);
await a.forEachAsync(e => console.log(e));

输出：

7
8
9

范例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining input elements
const a = 
  tf.data.array([4, 5, 6, 7, 8, 9]).skip(4);
await a.forEachAsync(e => console.log(e));

输出：

8
9

参考：https://js.tensorflow.org/api/latest/#tf.data.Dataset.skip

相关用法

注：本文由纯净天空筛选整理自arorakashish0911大神的英文原创作品 Tensorflow.js tf.data.Dataset.skip() Method。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。