Tensorflow.js tf.linspace()函数
Tensorflow.js 是一个由谷歌开发的开源库,用于在浏览器或节点环境中运行机器学习模型以及深度学习神经网络。
.linspace()函数用于在规定的时间间隔内查找规则距离的一系列数字。
句法 :
tf.linspace(start, stop, num)
参数:
- start:它是要生成的系列的规定起始值,类型为 number。
- stop:它是要生成的系列的规定结束值,并且是数字类型。
- num:它是要生成的值的规定数量,并且是 number 类型。
返回值:返回 tf.Tensor1D 对象。
示例 1:
Javascript
// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
// Defining all the parameters
var strt = 3;
var stp = 20;
var num = 5;
// Calling linspace() method and
// Printing output
tf.linspace(strt, stp, num).print();
Javascript
// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
// Calling linspace() method and
// Printing output
tf.linspace(12.4, 29.7, 5.4).print();
输出:
Tensor
[3, 7.25, 11.5, 15.75, 20]
示例 2:对所有声明的参数使用浮点值。
Javascript
// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
// Calling linspace() method and
// Printing output
tf.linspace(12.4, 29.7, 5.4).print();
输出:
Tensor
[12.3999996, 16.3318176, 20.2636356, 24.1954536, 28.1272717]
参考: https://js.tensorflow.org/api/latest/#linspace