Differences with torch.log2

View Source On Gitee

torch.log2

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

For more information, see torch.log2.

mindspore.ops.log2

mindspore.ops.log2(input) -> Tensor

For more information, see mindspore.ops.log2.

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 on CPU platform, while the dtype of the parameter input can be float16 or float32 on Ascend platform.

Categories

Subcategories

PyTorch

MindSpore

Difference

Parameters

Parameter 1

input

input

Both are Tensor, and the dtype of the parameter input in torch.log2 can be int or float. The dtype of the parameter input in mindspore.ops.log2 can be float16, float32, float64 on CPU platform, while float16 or float32 on Ascend platform.

Parameter 2

out

-

For detailed, refer to General Difference Parameter Table.