深度学习 nd.arange()

data_structure_wr2024-10-20 19:55

在 MXNet 中，nd.arange() 是一个用于生成均匀间隔的数值数组的函数。它的功能类似于 NumPy 的 arange() 函数。下面是 nd.arange() 的一些基本用法：

语法

python 复制代码

mx.nd.arange(start, stop=None, step=1, dtype=None, ctx=None)

参数

start：可选，开始值（包括）。
stop：结束值（不包括）。
step：步长，默认为 1。
dtype：数据类型，指定生成数组的数据类型。
ctx：上下文，指定计算设备（如 CPU 或 GPU）。

示例

python 复制代码

import mxnet as mx

# 生成从 0 到 9 的数组
arr = mx.nd.arange(10)
print(arr)  # 输出: [0. 1. 2. 3. 4. 5. 6. 7. 8. 9.]

# 生成从 1 到 10 的数组
arr2 = mx.nd.arange(1, 10)
print(arr2)  # 输出: [1. 2. 3. 4. 5. 6. 7. 8. 9.]

# 生成从 0 到 10 的数组，步长为 2
arr3 = mx.nd.arange(0, 10, 2)
print(arr3)  # 输出: [0. 2. 4. 6. 8.]