Python PyTorch zeros_like用法及代碼示例

本文簡要介紹python語言中 torch.zeros_like 的用法。

用法:

torch.zeros_like(input, *, dtype=None, layout=None, device=None, requires_grad=False, memory_format=torch.preserve_format) → Tensor

參數：

input(Tensor) -input 的大小將決定輸出張量的大小。

關鍵字參數：

• dtype(torch.dtype， 可選的) -返回張量的所需數據類型。默認值：如果 None ，則默認為 input 的數據類型。

• layout(torch.layout， 可選的) -返回張量的所需布局。默認值：如果 None ，則默認為 input 的布局。

• device(torch.device， 可選的) -返回張量的所需設備。默認值：如果 None ，則默認為 input 的設備。

• requires_grad(bool,可選的) -如果 autograd 應該在返回的張量上記錄操作。默認值：False。

• memory_format(torch.memory_format， 可選的) -返回的張量所需的內存格式。默認值：torch.preserve_format。

返回一個用標量值 0 填充的張量，其大小與 input 相同。 torch.zeros_like(input) 等同於 torch.zeros(input.size(), dtype=input.dtype, layout=input.layout, device=input.device) 。

警告

從 0.4 開始，此函數不支持 out 關鍵字。作為替代方案，舊的 torch.zeros_like(input, out=output) 等效於 torch.zeros(input.size(), out=output) 。

例子：

>>> input = torch.empty(2, 3)
>>> torch.zeros_like(input)
tensor([[ 0.,  0.,  0.],
        [ 0.,  0.,  0.]])