torch.Tensor.new_ones

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

返回一个大小为size的张量，并用1填充。默认情况下，返回的张量具有与当前张量相同的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.tensor((), dtype=torch.int32)
>>> tensor.new_ones((2, 3))
tensor([[ 1,  1,  1],
        [ 1,  1,  1]], dtype=torch.int32)