admin 管理员组

文章数量: 1086019


2024年3月26日发(作者:什么叫做mvc模式)

pytorch中range的用法

在PyTorch中,range()函数可以用于创建一个整数序列的张量。

以下是range()函数的用法示例:

1. 创建一个整数序列张量:

```python

import torch

# 创建一个从0到9的整数序列张量

x = (10)

print(x)

# 输出: tensor([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

```

2. 创建一个指定范围的整数序列张量:

```python

import torch

# 创建一个从2到9的整数序列张量

x = (2, 10)

print(x)

# 输出: tensor([2, 3, 4, 5, 6, 7, 8, 9])

```

3. 创建一个指定步长的整数序列张量:

```python

import torch

# 创建一个从0到9的步长为2的整数序列张量

x = (0, 10, 2)

print(x)

# 输出: tensor([0, 2, 4, 6, 8])

```

注意:range()函数创建的张量不包含结束值,即

半开区间。

[start, end)的


本文标签: 整数 创建 序列 叫做 指定