torch.randn_like()详解

torch.randn_like()是一个 PyTorch 函数，它返回一个与输入张量大小相同的张量，其中填充了均值为 0 方差为 1 的正态分布的随机值。

以下是 torch.randn_like()的语法：

output_tensor = torch.randn_like(input_tensor, dtype=None, layout=None, device=None, requires_grad=False)

其中

• input_tensor（必需）- 其大小将用于生成输出张量的输入张量。
• dtype（可选）- 输出张量所需的数据类型。默认为None，这意味着将使用输入张量的数据类型。
• layout（可选）- 输出张量所需的内存布局。默认为None，这意味着将使用输入张量的内存布局。
• device（可选）- 输出张量所需的设备。默认为None，这意味着将使用输入张量的设备。
• requires_grad（可选）- 输出张量是否应该在反向传播期间计算其梯度。默认为False。

下面是一个使用示例torch.randn_like()：

import torch

x = torch.randn(2, 3)
y = torch.randn_like(x)

print("x:")
print(x)

print("y:")
print(y)

输出结果表示：

x:
tensor([[-1.2325,  1.2024, -1.3687],
        [-0.9878, -0.3169,  2.3081]])
y:
tensor([[-0.4256, -0.7590, -0.2116],
        [ 1.0796, -0.0953,  0.0863]])

此时x与y的大小都是（2，3）