Tensorflow.js中的tf.randomGamma()函数

Tensorflow.js是一个基于JavaScript的深度学习库，它提供了许多函数来进行神经网络模型的搭建、训练和预测等。

tf.randomGamma()函数是其中的一个随机数生成函数，它可以生成遵循Gamma分布的随机数。

语法

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

参数说明

• shape: 用于指定生成随机数的形状，可以是一个数字表示生成一维数组，也可以是一个数组表示生成多维数组。例如，[2, 3]表示生成一个2行3列的数组。
• alpha: 用于指定Gamma分布中的形状参数，它必须为正实数。
• beta: 用于指定Gamma分布中的比例参数，它必须为正实数。
• dtype: 用于指定生成随机数的类型，可以是'float32'、'int32'、'bool'等。

返回值

该函数返回一个张量，形状与shape参数指定的一致，元素为生成的Gamma分布的随机数。

使用示例

// 生成一个遵循Gamma分布的随机数，形状为[2, 3]，alpha值为2，beta值为1，类型为float32。
const result = tf.randomGamma([2, 3], 2, 1, 'float32');
result.print();

输出结果如下：

Tensor
  [[1.1973699  0.7779945  2.129099  ]
   [2.1574786  4.1065426  4.283812  ]]