Python PyTorch empty_like用法及代碼示例

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

用法:

torch.empty_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。

返回與 input 大小相同的未初始化張量。 torch.empty_like(input) 等同於 torch.empty(input.size(), dtype=input.dtype, layout=input.layout, device=input.device) 。

例子：

>>> torch.empty((2,3), dtype=torch.int64)
tensor([[ 9.4064e+13,  2.8000e+01,  9.3493e+13],
        [ 7.5751e+18,  7.1428e+18,  7.5955e+18]])