Differences with torch.empty

View Source On Gitee

torch.empty

torch.empty(
    *size,
    *,
    out=None,
    dtype=None,
    layout=torch.strided,
    device=None,
    requires_grad=False,
    pin_memory=False,
    memory_format=torch.contiguous_format
) -> Tensor

For more information, see torch.empty.

mindspore.numpy.empty

mindspore.numpy.empty(shape, dtype=mstype.float32) -> Tensor

For more information, see mindspore.numpy.empty.

Differences

PyTorch: Return an uninitialized tensor, the shape of which is defined by size.

MindSpore: MindSpore API basically implements the same function as PyTorch, but the default value of the dtype parameter is different.

Categories

Subcategories

PyTorch

MindSpore

Difference

Parameters

Parameter 1

size

shape

Same function, different parameter names

Parameter 2

out

-

Not involved

Parameter 3

dtype

dtype

Same function, different default values

Parameter 4

layout

-

Not involved

Parameter 5

device

-

Not involved

Parameter 6

requires_grad

-

MindSpore does not have this parameter and supports reverse derivation by default

Parameter 7

pin_memory

-

Not involved

Parameter 8

memory_format

-

Not involved

Code Example 1

For the parameter dtype, PyTorch defaults to None, and the output type is torch.float32, while MindSpore defaults to mstype.float32.

# PyTorch
import torch

torch_output = torch.empty(2, 3)
print(list(torch_output.shape))
# [2, 3]

# MindSpore
import mindspore

ms_output = mindspore.numpy.empty((2, 3))
print(ms_output.shape)
# (2, 3)