Python – PyTorch log() 方法

简介

在PyTorch中，torch.log()函数是用于计算输入张量的自然对数的方法。它返回一个新张量，其中每个元素等于输入张量的对数值。

语法

torch.log(input, out=None)

参数

• input (Tensor)：输入张量。

• out (Tensor, 可选)：输出张量，用于存储结果。

返回值

返回一个新的张量，其中每个元素等于输入张量的对数值。

示例

import torch

# 创建一个输入张量
x = torch.tensor([1, 10, 100, 1000], dtype=torch.float32)

# 计算对数
result = torch.log(x)

# 打印结果
print(result)

输出结果：

tensor([0.0000, 2.3026, 4.6052, 6.9078])

注意事项

• 输入张量的所有元素必须大于零，否则会抛出运行时错误。
• 如果要在原地计算自然对数，可以使用torch.log_()方法。
• 如果需要计算以其他对数为底的对数，可以使用torch.log10()（以10为底）或torch.log2()（以2为底）方法。

以上就是Python中PyTorch库的log()方法的介绍。该方法用于计算输入张量的自然对数，并返回一个新的张量。