Differences with torch.log2

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.