Differences with torch.floor

View Source On Gitee

torch.floor

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

For more information, see torch.floor.

mindspore.ops.floor

mindspore.ops.floor(input) -> Tensor

For more information, see mindspore.ops.floor.

Differences

MindSpore API function is consistent with that of PyTorch, with differences in the data types supported by the parameters.

PyTorch: The dtype of the parameter input can be int or float.

MindSpore: The dtype of the parameter input can be float16, float32, or float64.

Categories

Subcategories

PyTorch

MindSpore

Difference

Parameters

Parameter 1

input

input

Both are Tensor, and the dtype of the parameter input in torch.floor can be int or float. The dtype of the parameter input in mindspore.ops.floor can be float16, float32 or float64.

Parameter 2

out

-

For detailed, refer to General Difference Parameter Table.