mindspore.Tensor.exponential_

查看源文件
mindspore.Tensor.exponential_(lambd=1, *, generator=None)[源代码]

根据指数分布生成随机数填充Tensor。

\[f(x) = \lambda \exp(-\lambda x)\]

警告

  • 仅支持 Atlas A2 训练系列产品。

  • 这是一个实验性API,后续可能修改或删除。

参数:
  • lambd (float, 可选) - 指数分布的参数。默认值: 1

关键字参数:
  • generator (Generator, 可选) - 随机数生成器。默认值: None

返回:

Tensor,shape和数据类型与输入相同。

支持平台:

Ascend

样例:

>>> import mindspore
>>> x = mindspore.Tensor([1, 2, 3.0])
>>> out = x.exponential_(2)
>>> print(out.shape)
(3,)