torch.randn(batch_size, in_channels, w, h)
时间: 2023-11-16 10:07:55 浏览: 218
`torch.randn(batch_size, in_channels, w, h)` 会生成一个指定形状的随机张量。其中 `batch_size` 表示批次大小,`in_channels` 表示输入通道数,`w` 和 `h` 则表示张量的宽度和高度。这个函数会返回一个张量,其中的元素都是从标准正态分布中随机抽取的,具有指定的形状。
例如,如果我们想要生成一个批次大小为 2,输入通道数为 3,宽度为 4,高度为 4 的张量,可以使用以下代码:
```
import torch
x = torch.randn(2, 3, 4, 4)
print(x)
```
输出结果类似于:
```
tensor([[[[ 1.2271, 0.2421, -0.6105, -1.5488],
[-1.2785, -1.1008, -0.9254, 0.2425],
[-0.9050, -0.5843, -0.3160, 0.1477],
[-0.4255, 1.1630, -0.1759, 0.2668]],
[[ 0.9667, 1.0385, -0.0688, 0.4463],
[-1.6751, -1.0941, -0.2750, -0.9461],
[ 0.2135, 0.1377, 1.0522, -1.0206],
[-0.3786, -0.4150, -1.9548, 0.7163]],
[[-0.3319, -0.3414, -0.1942, -0.3945],
[-0.4025, -1.4332, 0.5941, -1.2622],
[-0.2395, -1.1825, -1.4708, -1.2992],
[-0.3179, 0.6087, -0.2520, 1.0131]]],
[[[-0.6207, -0.1085, -1.0115, -0.0556],
[-0.0735, -0.5308, -0.2507, 0.6339],
[-0.2363, -0.2707, -0.2864, -0.4051],
[-0.3477, -1.7979, 0.6153, 0.6357]],
[[-0.5134, -0.1967, -1.1536, 0.1384],
[-0.6141, -0.7290, 1.0909, -0.8485],
[-0.5628, -0.3050, 0.1641, 0.9212],
[-0.0648, 0.7800, -0.4631, -0.4183]],
[[ 0.8056, 0.0912, -0.6264, 0.1679],
[ 0.8461, -0.0488, 1.0714, -0.0591],
[-0.4256, 0.1218, -0.4453, -0.8878],
[-0.7765, -0.1209, -1.0712, -0.9805]]]])
```
其中的元素都是随机生成的,每次执行代码都会得到不同的结果。
阅读全文