torch::randn
时间: 2023-09-18 09:10:41 浏览: 130
pyTorch.randn()、rand()、cat()、pow()、scatter_()、.squeeze() 、.unsqueeze()
5星 · 资源好评率100%
`torch::randn` 是 PyTorch C++ API 中用于生成随机张量的函数之一,它可以生成一个张量,其中的元素服从均值为0,标准差为1的正态分布。
函数签名为:`torch::randn({size}, options)`。
其中,`{size}` 是一个张量的形状,如 `{3, 4}` 表示一个 $3\times 4$ 的张量。`options` 表示生成的张量的数据类型、设备等信息。例如,可以使用 `torch::kFloat32` 表示生成的张量的数据类型为 float32(即单精度浮点数),使用 `torch::kCUDA` 表示生成的张量存储在 GPU 上。
下面是一个示例代码,使用 `torch::randn` 生成一个形状为 `{2, 3}` 的随机张量,并打印出来:
```cpp
#include <torch/torch.h>
#include <iostream>
int main() {
auto options = torch::TensorOptions().dtype(torch::kFloat32);
auto x = torch::randn({2, 3}, options);
std::cout << "Random tensor:\n" << x << std::endl;
return 0;
}
```
阅读全文