📜  Tensorflow.js tf.range()函数

📅  最后修改于: 2022-05-13 01:56:48.062000             🧑  作者: Mango

Tensorflow.js tf.range()函数

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

tf。 range()用于创建一个新的tf.Tensor1D ,其中填充了在startstop、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()


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()


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()


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
    [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