比较与torch.ceil的功能差异

torch.ceil

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

更多内容详见torch.ceil

mindspore.ops.ceil

mindspore.ops.ceil(x) -> Tensor

更多内容详见mindspore.ops.ceil

差异对比

PyTorch:返回带有 input 元素的 ceil 的新张量,该元素大于或等于每个元素的最小整数。

MindSpore:MindSpore此API实现功能与PyTorch一致,仅参数名不同。

分类

子类

PyTorch

MindSpore

差异

参数

参数1

input

x

功能一致,参数名不同

参数2

out

-

不涉及

代码示例1

两API实现功能一致,用法相同。

# PyTorch
import numpy as np
import torch
from torch import tensor

input = torch.tensor(np.array([2.5, -1.5, 1, 1.4448, 0.5826]), dtype=torch.float32)
output = torch.ceil(input).numpy()
print(output)
# [ 3. -1.  1.  2.  1.]

# MindSpore
import numpy as np
import mindspore
import mindspore.ops as ops
from mindspore import Tensor

x = Tensor(np.array([2.5, -1.5, 1, 1.4448, 0.5826]), mindspore.float32)
output = ops.ceil(x).asnumpy()
print(output)
# [ 3. -1.  1.  2.  1.]