torch.Tensor.new_full

Tensor.new_full(size, fill_value, *, dtype=None, device=None, requires_grad=False, layout=torch.strided, pin_memory=False) → Tensor 修正后的翻译如下： Tensor.new_full(size, fill_value, *, dtype=None, device=None, requires_grad=False, layout=torch.strided, pin_memory=False) → Tensor

返回一个大小为size的张量，并用fill_value填充。默认情况下，返回的张量具有与当前张量相同的torch.dtype和torch.device。

参数

fill_value (标量) – 填充输出张量的数值。

关键字参数

• 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((2,), dtype=torch.float64)
>>> tensor.new_full((3, 4), 3.141592)
tensor([[ 3.1416,  3.1416,  3.1416,  3.1416],
        [ 3.1416,  3.1416,  3.1416,  3.1416],
        [ 3.1416,  3.1416,  3.1416,  3.1416]], dtype=torch.float64)