torch.rande
时间: 2023-08-19 08:17:27 浏览: 106
`torch.rand()` 是 PyTorch 中的一个函数,用于生成指定形状的随机张量,张量中的元素服从均匀分布在区间 [0,1) 内。
以下是使用 `torch.rand()` 的示例代码:
```pythonimport torch#生成一个形状为 (2,3) 的随机张量x = torch.rand(2,3)
print(x)
```
输出示例:
```
tensor([[0.7426,0.9443,0.4405],
[0.1118,0.6127,0.6385]])
```
在这个例子中,我们使用 `torch.rand(2,3)`生成了一个形状为 (2,3) 的随机张量。每个元素都是从均匀分布 [0,1) 中随机采样得到的。
需要注意的是,`torch.rand()` 函数生成的张量是在 CPU 上的,默认数据类型是 `torch.float32`。如果需要在 GPU 上操作,可以使用 `.cuda()` 方法将其转移到 GPU 上。同时,可以使用 `.to(dtype)` 方法指定数据类型,例如 `.to(torch.float16)` 将张量转换为半精度浮点数。
阅读全文