Python PyTorch log2() 方法

简介

log2() 是 PyTorch 中的一个函数，它可以计算输入张量中每个元素的以2为底的对数。具体可查看 PyTorch 官方文档

语法

torch.log2(input, out=None) -> Tensor

参数

• input (Tensor) – 输入张量。
• out (Tensor, optional) – 输出张量。

返回值

• 输出张量。返回一个包含输入张量每个元素的以2为底的对数的张量。输出张量的形状与输入张量相同。

示例

import torch

a = torch.tensor([1, 2, 4, 8])
b = torch.log2(a)
print(b)

输出结果:

tensor([ 0.,  1.,  2.,  3.])

解释

上述示例中，先创建一个输入张量 a，包含元素 [1, 2, 4, 8]。然后使用 log2() 函数对其进行操作，返回结果为 [0, 1, 2, 3]。这是因为 log2(1) = 0，log2(2) = 1，log2(4) = 2，log2(8) = 3。

总结

log2() 函数是 PyTorch 中的一个重要的数学操作函数之一。它可以方便地对输入张量进行操作，返回张量中每个元素的以2为底的对数。在使用时，需要注意输入张量应为浮点类型或者是整型。