torch.randint函数
时间: 2023-10-14 19:15:24 浏览: 52
torch.randint函数是一个PyTorch中的函数,用于生成指定形状的随机整数张量。其语法如下:
```
torch.randint(high, size, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor
```
其中,参数含义如下:
- high: 随机整数的上限(不包括该值)。
- size: 生成张量的形状。
- dtype: 生成张量的数据类型。
- layout: 生成张量的内存布局。
- device: 生成张量所在的设备。
- requires_grad: 是否需要自动求导。
例如,以下代码生成了一个形状为(2, 3)、值在[0, 10)范围内的随机整数张量:
```
import torch
x = torch.randint(10, size=(2, 3))
print(x)
```
输出结果如下:
```
tensor([[7, 0, 9],
[9, 1, 0]])
```
相关问题
torch.randint
`torch.randint` 是 PyTorch 中的一个函数,用于生成指定大小和范围的整数张量。其语法如下:
```
torch.randint(high, size, dtype=None, layout=torch.strided, device=None, requires_grad=False) -> Tensor
```
其中,
- `high`:生成的整数张量的上限值(不包括)。
- `size`:生成的整数张量的形状。
- `dtype`:生成的整数张量的数据类型,默认为 `torch.int64`。
- `layout`:生成的整数张量的布局,默认为 `torch.strided`。
- `device`:生成的整数张量的设备,默认为当前设备。
- `requires_grad`:生成的整数张量是否需要梯度,默认为 `False`。
例如,生成一个形状为 (3, 2)、值域为 [0, 5) 的整数张量,可以使用以下代码:
```python
import torch
x = torch.randint(5, (3, 2))
print(x)
```
输出:
```
tensor([[4, 3],
[3, 4],
[1, 0]])
```
torch.randint_like
`torch.randint_like`函数可以生成与输入张量形状相同的随机整数张量。可以指定生成整数的范围和数据类型。下面是一个使用`torch.randint_like`函数的例子:
```python
import torch
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
y = torch.randint_like(x, low=0, high=10, dtype=torch.int32)
print(y)
```
上述代码中,我们首先创建了一个形状为`(2, 3)`的张量`x`,然后使用`torch.randint_like`函数生成了一个与`x`形状相同的随机整数张量`y`,其中整数的范围是`[0, 10)`,数据类型是`torch.int32`。最后打印输出了生成的随机整数张量`y`。
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)