nvidia.dali.fn.random.uniform#

nvidia.dali.fn.random.uniform(__shape_like=None, /, *, bytes_per_sample_hint=[0], dtype=None, preserve=False, range=[-1.0, 1.0], seed=-1, shape=None, values=None, device=None, name=None)#

生成服从均匀分布的随机数。

它可以配置为生成 range [min, max) 范围内的连续均匀分布，或者生成离散均匀分布，其中指定的 values [v0, v1, …, vn] 中的任何值以相等的概率出现。

生成数据的形状可以显式地使用 shape 参数指定，或者选择与提供的 __shape_like 输入的形状匹配。如果两者都不存在，则每个样本生成一个值。

支持的后端

  • ‘cpu’

  • ‘gpu’

参数:

__shape_like (TensorList, 可选) – 如果提供，则此输入的形状将用于推断输出的形状。

关键字参数:

  • bytes_per_sample_hint (int 或 list of int, 可选, 默认 = [0]) –

    每个样本的输出大小提示（以字节为单位）。

    如果指定，则驻留在 GPU 或页锁定主机内存中的运算符输出将被预先分配，以容纳此大小的样本批次。

  • dtype (nvidia.dali.types.DALIDataType, 可选) –

    输出数据类型。

    注意

    生成的数字将转换为输出数据类型，并在必要时进行舍入和钳位。

  • preserve (bool, 可选, 默认 = False) – 即使运算符的输出未使用，也阻止将其从图中删除。

  • range (float 或 list of float 或 TensorList of float, 可选, 默认 = [-1.0, 1.0]) –

    连续均匀分布的范围 [min, max)。

    此参数与 values 互斥。

    警告

    当指定整数类型作为 dtype 时，由于舍入，生成的数字可能会超出指定的范围。

  • seed (int, 可选, 默认 = -1) – 随机种子；如果未设置，将自动分配一个。

  • shape (int 或 list of int 或 TensorList of int, 可选) – 输出数据的形状。

  • values (float 或 list of float 或 TensorList of float, 可选) –

    离散均匀分布产生的离散值 [v0, v1, …, vn]。

    此参数与 range 互斥。

在本页上