快捷方式

torch.Tensor.geometric_

Tensor.geometric_(p, *, generator=None) Tensor

用从几何分布中提取的元素填充self张量

P(X=k)=(1p)k1p,k=1,2,...P(X=k) = (1 - p)^{k - 1} p, k = 1, 2, ...

注意

torch.Tensor.geometric_()k 次试验是第一次成功,因此在 {1,2,}\{1, 2, \ldots\} 中抽取样本,而 torch.distributions.geometric.Geometric()(k+1)(k+1) 次试验是第一次成功,因此在 {0,1,}\{0, 1, \ldots\} 中抽取样本。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取适合初学者和高级开发者的深度教程

查看教程

资源

查找开发资源并解答您的问题

查看资源