Tensorflow.js tf.range()函数

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

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