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


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


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

tf.randomGamma()函数用于创建tf.Tensor,该值具有从伽马分布中采样的值。

用法:

tf.randomGamma(shape, alpha, beta, dtype, seed)

参数:此函数接受三个参数,如下所示:

  • shape:定义输出张量形状的整数数组。
  • alpha:伽玛分布的形状参数。
  • beta:这是一个可选参数。伽玛分布的反比例参数。预设值是1。
  • dtype:输出的数据类型。可能的数据类型值为‘float32’或‘int32’。这也是一个可选参数。
  • seed:它是一个可选参数。随机数生成器的种子。

返回:返回tf.Tensor



范例1:

Javascript


// Importting the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating the tensor with values sampled 
// from a gamma distribution
const x=tf.randomGamma([5], 0);
  
// Printing the tensor
x.print();

输出:

Tensor
    [0, 0, 0, 0, 0]

范例2:

Javascript


// Importting the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating the tensor with values sampled 
// from a gamma distribution
const x=tf.randomGamma([5], 1);
  
// Printing the tensor
x.print();

输出:

Tensor
    [1.4808178, 1.6668015, 0.9527208, 1.6024575, 1.6021353]

范例3:

Javascript


// Importting the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating the tensor with values sampled
// from a gamma distribution
const x=tf.randomGamma([2,2], 1);
  
// Printing the tensor
x.print();

输出:

Tensor
    [[0.1157758, 1.4427431],
     [0.4978852, 0.1617882]]

范例4:

Javascript


// Importting the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Creating the tensor with values sampled 
// from a gamma distribution
const x=tf.randomGamma([5], 1,2,'int32',98);
  
// Printing the tensor
x.print();

输出:

Tensor
    [0, 1, 4, 0, 1]

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

相关用法


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