torch.ceil() 介绍
时间: 2023-08-21 17:07:36 浏览: 48
Pytorch maxpool的ceil_mode用法
`torch.ceil()` 是 PyTorch 中的一个函数,用于计算输入张量中每个元素的上限值。也就是说,将输入张量中的每个元素向上取整,并返回结果张量,其形状与输入张量相同。
该函数的语法格式如下:
```
torch.ceil(input, out=None) -> Tensor
```
其中,参数 `input` 是输入张量,参数 `out` 是可选的输出张量。如果指定了 `out`,则将计算结果存储在 `out` 张量中,并返回 `out` 张量。否则,将创建一个新的张量来存储计算结果,并返回该张量。
以下示例代码演示了 `torch.ceil()` 函数的用法:
```python
import torch
x = torch.tensor([-1.5, 2.7, 3.8, -4.2])
y = torch.ceil(x)
print(y)
```
运行结果如下:
```
tensor([-1., 3., 4., -4.])
```
可以看到,`torch.ceil()` 函数将输入张量 `x` 中的每个元素向上取整,并返回结果张量 `y`。
阅读全文