torch.Tensor.new_empty

Tensor.new_empty(size, *, dtype=None, device=None, requires_grad=False, layout=torch.strided, pin_memory=False) → Tensor

返回一个大小为 size 的张量，填充未初始化的数据。默认情况下，返回的张量具有与当前张量相同的 torch.dtype 和 torch.device。

参数

size (int...) – 一个列表、元组或 torch.Size，包含定义输出张量形状的整数。

关键字参数

• dtype (torch.dtype) - 返回的张量所需的类型。默认值：如果为 None，则与此张量相同的 torch.dtype。

• device (torch.device, 可选) – 返回的张量所需的设备。默认情况下，如果为 None，则与该张量相同的 torch.device。

• requires_grad (bool, optional) – 是否启用自动求导记录返回的张量上的操作。默认值：False。

• layout (torch.layout, 可选) - 指定返回张量的布局。默认值为 torch.strided。

• pin_memory (bool, 可选) – 如果启用，返回的张量将在固定内存中分配。仅适用于 CPU 张量。默认值： False。

示例：

>>> tensor = torch.ones(())
>>> tensor.new_empty((2, 3))
tensor([[ 5.8182e-18,  4.5765e-41, -1.0545e+30],
        [ 3.0949e-41,  4.4842e-44,  0.0000e+00]])