当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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


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

.sigmoid()函数用于查找指定张量输入的S形,即1 /(1 + exp(-x)),并逐个元素地完成。

用法:

tf.sigmoid(x)

参数:该函数接受单个参数,如下所示:

  • x:它是张量输入,可以是tf.Tensor,TypedArray或Array类型。

返回值:它返回tf.Tensor对象。



范例1:

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining tensor input elements
const y = tf.tensor1d([-1, 15, 0, Math.E-1]);
  
// Calling sigmoid() method and
// printing output
y.sigmoid().print();

输出:

Tensor
    [0.2689414, 0.9999996, 0.5, 0.8479074]

范例2:在此示例中,参数直接传递到Sigmoid函数。

Javascript


// Importing the tensorflow.js library 
import * as tf from "@tensorflow/tfjs"
  
// Defining float values
var val = [0.5, -1.5, .66];
  
// Calling tensor1d method
const y = tf.tensor1d(val);
  
// Calling sigmoid() method
var res = tf.sigmoid(y)
  
// Printing output
res.print();

输出:

Tensor
    [0.6224594, 0.1824255, 0.6592603]

相关用法


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