torch.randn_like()函數

torch.randn_like(input, *, dtype=None, layout=None, device=None, requires_grad=False, memory_format=torch.preserve_format) -> Tensor

 返回一個和輸入大小相同的張量，其由均值為0、方差為1的標准正態分布填充。

即torch.randn_like(input)等價於torch.randn(input.size(), dtype=input.dtype, layout=input.layout, device=input.device)

dtype（torch.dtype, optional）期望返回張量的類型，默認為input的類型
layout（torch.layout, optional）期望返回張量的分布，默認為input的分布
device（torch.device, optional）期望返回張量使用的設備，默認與input相同
requires_grad（bool, optional）如果需要在返回張量上保存梯度，默認為False
memory_format（torch.memory_format, optional）期望返回張量的保存類型，默認為torch.preserve_format