Tensorflow.js tf.range()用法及代码示例

Tensorflow.js是Google开发的开源库，用于在浏览器或节点环境中运行机器学习模型和深度学习神经网络。

tf。 range()用于创建一个新的tf.Tensor1D，其中填充了借助start，stop，step和dtype提供的范围内的数字。

用法:

tf.range(start, stop, step, dtype)

参数：

start:它是一个整数起始值，表示范围的起始编号。
stop:它是一个整数终止值，表示范围的结束号，并且不包括在内。
step:它是一个整数增量，默认情况下为1或-1。它是一个可选参数。
dtype:它是输出张量的数据类型。默认为‘float32’。它是一个可选参数。

返回值：它返回一个新的Tensor1D对象。

范例1：在此示例中，我们尝试使用默认步骤1生成一个范围为1到9的数字。

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating the tensor1D array by tf.range
var val = tf.range(1,10);
  
// Printing the tensor
val.print()

输出：

Tensor
    [1, 2, 3, 4, 5, 6, 7, 8, 9]

范例2：在此示例中，我们尝试使用步长2生成1到10之间的奇数。

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating the tensor1D array by tf.range
var val = tf.range(1,10,2);
  
// Printing the tensor
val.print()

输出：

Tensor
    [1, 3, 5, 7, 9]

范例3：在此示例中，我们尝试使用步长2生成0到10之间的偶数。

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating the tensor1D array by tf.range
var val = tf.range(0,10,2);
  
// Printing the tensor
val.print()

输出：

Tensor
    [0, 2, 4, 6, 8],

范例4：在此示例中，我们将尝试使用dtype参数。

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating the tensor1D array by tf.range
var val = tf.range(-1,1,1,'bool');
  
// Printing the tensor
val.print()

输出：

Tensor
    [true, false]

参考：https://js.tensorflow.org/api/latest/#range

相关用法

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