torch.Tensor.new_ones

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

返回一個大小為 size 且填充了 1 的 Tensor。預設情況下，返回的 Tensor 與此 Tensor 具有相同的 torch.dtype 和 torch.device。

引數

size (int...) – 定義輸出張量形狀的整數列表、元組或 torch.Size。

關鍵字引數

• dtype (torch.dtype, 可選) – 返回張量的期望型別。預設值：如果為 None，則與此張量具有相同的 torch.dtype。

• device (torch.device, 可選) – 返回張量的期望裝置。預設值：如果為 None，則與此張量具有相同的 torch.device。

• requires_grad (bool, 可選) – 如果 autograd 應該記錄返回張量上的操作。預設值：False。

• layout (torch.layout, 可選) – 返回 Tensor 的期望佈局。預設值：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)