比较与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.]