本文简要介绍python语言中 torch.bincount
的用法。
用法:
torch.bincount(input, weights=None, minlength=0) → Tensor
一个形状为
Size([max(input) + 1])
的张量,如果input
不为空,否则为Size(0)
输出(Tensor)
计算非负整数数组中每个值的频率。
bin 的数量(大小 1)比
input
中的最大值大一,除非input
为空,在这种情况下,结果是大小为 0 的张量。如果指定了minlength
,则 bin 的数量为至少minlength
并且如果input
为空,则结果是大小为minlength
的张量用零填充。如果n
是位置i
的值,如果指定了weights
则为out[n] += weights[i]
,否则为out[n] += 1
。注意
当给定 CUDA 设备上的张量时,此操作可能会产生不确定的梯度。有关详细信息,请参阅重现性。
例子:
>>> input = torch.randint(0, 8, (5,), dtype=torch.int64) >>> weights = torch.linspace(0, 1, steps=5) >>> input, weights (tensor([4, 3, 6, 3, 4]), tensor([ 0.0000, 0.2500, 0.5000, 0.7500, 1.0000]) >>> torch.bincount(input) tensor([0, 0, 0, 2, 2, 0, 1]) >>> input.bincount(weights) tensor([0.0000, 0.0000, 0.0000, 1.0000, 1.0000, 0.0000, 0.5000])
参数:
返回:
返回类型:
相关用法
- Python PyTorch binary_cross_entropy用法及代码示例
- Python PyTorch binary_cross_entropy_with_logits用法及代码示例
- Python PyTorch bitwise_right_shift用法及代码示例
- Python PyTorch bitwise_and用法及代码示例
- Python PyTorch bitwise_not用法及代码示例
- Python PyTorch bitwise_xor用法及代码示例
- Python PyTorch bitwise_or用法及代码示例
- Python PyTorch bitwise_left_shift用法及代码示例
- Python PyTorch bernoulli用法及代码示例
- Python PyTorch backward用法及代码示例
- Python PyTorch batched_powerSGD_hook用法及代码示例
- Python PyTorch bleu_score用法及代码示例
- Python PyTorch broadcast_tensors用法及代码示例
- Python PyTorch build_vocab_from_iterator用法及代码示例
- Python PyTorch broadcast_object_list用法及代码示例
- Python PyTorch broadcast_shapes用法及代码示例
- Python PyTorch bf16_compress_wrapper用法及代码示例
- Python PyTorch bmm用法及代码示例
- Python PyTorch baddbmm用法及代码示例
- Python PyTorch broadcast_to用法及代码示例
- Python PyTorch bf16_compress_hook用法及代码示例
- Python PyTorch bucketize用法及代码示例
- Python PyTorch block_diag用法及代码示例
- Python PyTorch frexp用法及代码示例
- Python PyTorch jvp用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.bincount。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。