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


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM