Python | PyTorch cos() 方法

在 PyTorch 中，cos() 方法可以用于计算张量中每个元素的余弦值。它是 torch 模块中的一个函数，其中我们可以调用任何张量以获得该张量中每个元素的余弦值。

语法

torch.cos(input, out=None)

参数

| 参数 | 描述 | |----------|-------------| | input | 输入张量 | | out | 输出张量 |

返回值

有时候，我们需要计算输入张量中每个元素的余弦值，这时候我们可以使用 PyTorch 中的 cos() 方法。它可以将每个元素除以 180^o 后的余弦值作为新张量的值。这个新张量是与输入张量大小相同的张量。

代码示例

import torch

# 定义一个 2x2 的张量
x = torch.tensor([[0, 30], [45, 60]], dtype=torch.float)
# 计算余弦值
y = torch.cos(x)
print(y)

输出结果为：

tensor([[ 1.0000,  0.1543],
        [ 0.5253,  0.5000]])

这里，我们首先定义了一个 2x2 的张量，并将其作为输入张量传递给 cos() 方法。方法返回新张量 y，其中每个元素都是输入张量中对应元素的余弦值。

PyTorch 中的 cos() 方法还有一个可选参数 out。我们可以将其用于计算结果的输出张量。如果指定了 out，那么 cos() 方法将使用该张量作为输出，而不是创建一个新的输出张量。如果未指定 out，那么方法将生成新的输出张量。

import torch

# 定义一个 2x2 的张量
x = torch.tensor([[0, 30], [45, 60]], dtype=torch.float)
# 创建一个新的张量来存储结果
out = torch.tensor([[], []], dtype=torch.float)
# 计算余弦值
torch.cos(x, out=out)
print(out)

输出结果为：

tensor([[ 1.0000,  0.1543],
        [ 0.5253,  0.5000]])

这里，我们定义了一个 2x2 的张量 x，并创建了一个空的输出张量 out。使用 cos() 方法计算余弦值，并将其存储在 out 张量中。输出结果与上面的示例相同。

总之，PyTorch 中的 cos() 方法可以用于计算输入张量中每个元素的余弦值。我们还可以指定一个输出张量，以避免创建新的张量。