pytorch :sampler里的随机int

int(torch.empty((), dtype=torch.int64).random_().item())

文档里是这么描述.random_()的:

 

"""
random_(from=0, to=None, *, generator=None) -> Tensor

Fills :attr:self tensor with numbers sampled from the discrete uniform
distribution over [from, to - 1]. If not specified, the values are usually
only bounded by :attr:self tensor's data type. However, for floating point
types, if unspecified, range will be [0, 2^mantissa] to ensure that every
value is representable. For example, torch.tensor(1, dtype=torch.double).random_()
will be uniform in [0, 2^53].
"""

posted @ 2021-04-01 10:00  说分手后还能做炮友?  阅读(114)  评论(0)    收藏  举报