tf.random.uniform

323 阅读1分钟

tf.random.uniform 是 TensorFlow 的一个函数,用于生成具有均匀分布的随机数。

函数的签名如下:

tf.random.uniform(shape, minval=0, maxval=None, dtype=tf.float32, seed=None, name=None)

参数说明:

  • shape:一个整数或列表,表示要生成的随机张量的形状。
  • minval:一个浮点数,表示生成随机数的最小值。默认为0。
  • maxval:一个浮点数,表示生成随机数的最大值。如果未指定,则默认为1。
  • dtype:一个数据类型,表示生成随机数的数据类型。默认为tf.float32
  • seed:一个整数,表示随机数生成器的种子。如果指定了种子,则每次生成的随机数序列都是相同的。如果未指定种子,则使用系统时间作为默认种子。
  • name:一个字符串,表示操作的名称。

函数返回一个具有指定形状、最小值和最大值的均匀分布的随机张量。

下面是一个使用示例:

import tensorflow as tf

# 生成一个形状为 (2, 2) 的均匀分布随机张量,最小值为0,最大值为1
x = tf.random.uniform((2, 2))

# 生成一个形状为 (2, 2) 的均匀分布随机张量,最小值为-1,最大值为0
y = tf.random.uniform((2, 2), minval=-1, maxval=0)

# 打印结果
print(x)
print(y)

这个示例中,我们使用tf.random.uniform函数生成了两个形状为(2, 2)的均匀分布随机张量。其中,张量x的最小值为0,最大值为1,张量y的最小值为-1,最大值为0。最后,我们打印了这两个张量的值。由于生成的是随机数,每次运行代码时输出的结果都可能不同。