torch.ceil
时间: 2023-07-03 08:06:35 浏览: 160
Pytorch maxpool的ceil_mode用法
torch.ceil() 是一个 torch 模块中的函数,它用于将输入张量中的每个元素向上取整到最近的整数。它的语法如下:
```python
torch.ceil(input, out=None) -> Tensor
```
其中,`input` 是输入张量,`out` 是输出张量(可选参数)。如果指定了 `out` 参数,则结果将会被写入到 `out` 张量中,否则将会创建一个新的张量来保存结果。
以下是一个使用 torch.ceil() 的示例:
```python
import torch
x = torch.tensor([1.4, 2.6, 3.1, 4.9])
y = torch.ceil(x)
print(y) # tensor([2., 3., 4., 5.])
```
在这个示例中,我们创建了一个包含四个浮点数的张量 `x`,然后使用 `torch.ceil()` 将其向上取整到最近的整数,并将结果保存在张量 `y` 中。最后,我们打印出了张量 `y` 的内容。
阅读全文