Python - PyTorch ceil() 方法

在 PyTorch 中，我们可以使用 ceil() 方法来获取输入张量中每个元素的上取整值。

语法

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

• input：输入张量。
• out：可选参数，输出张量。

示例

import torch

x = torch.tensor([-1.5, 2.7, 0.3, -4.2])
y = torch.ceil(x)
print(y)

输出结果为：

tensor([-1.,  3.,  1., -4.])

解释

在上面的示例中，我们首先创建了一个具有四个元素的张量x，其中包含一些正数和负数。

然后，我们使用 ceil() 方法对 x 进行操作，该方法会将 x 中的每个元素上取整，然后返回一个新的张量 y。

最后，我们打印 y，结果显示 y 中的每个元素都是 x 中对应元素的上取整值。

注意，-1.5 的上取整值为 -1.0，而并非 0.0。

结论

在这篇文章中，我们介绍了 PyTorch 中的 ceil() 方法，它可以用来获取输入张量中每个元素的上取整值。该方法对于需要进行数学计算的深度学习应用程序非常有用。