PyTorch 中的 linspace

百变鹏仔 5天前 #Python

文章标签 PyTorch

请我喝杯咖啡☕

本篇博文将介绍 PyTorch 中的 linspace() 函数，它用于创建在起始值和结束值之间均匀分布的一维张量。

备忘录：

linspace() 函数可以生成包含零个或多个均匀间隔的整数、浮点数或复数元素的一维张量。

备忘录：

以下是一些示例代码：

import torchprint(torch.linspace(start=10, end=20, steps=0))  # tensor([])print(torch.linspace(start=20, end=10, steps=0))  # tensor([])print(torch.linspace(start=10., end=20., steps=1))  # tensor([10.])print(torch.linspace(start=20, end=10, steps=1))  # tensor([20.])print(torch.linspace(start=10., end=20., steps=2))  # tensor([10., 20.])print(torch.linspace(start=20, end=10, steps=2))  # tensor([20., 10.])print(torch.linspace(start=10., end=20., steps=3))  # tensor([10., 15., 20.])print(torch.linspace(start=20, end=10, steps=3))  # tensor([20., 15., 10.])print(torch.linspace(start=10., end=20., steps=4))  # tensor([10.0000, 13.3333, 16.6667, 20.0000])print(torch.linspace(start=20., end=10., steps=4))  # tensor([20.0000, 16.6667, 13.3333, 10.0000])print(torch.linspace(start=10, end=20, steps=4, dtype=torch.int64)) # 注意：结果仍然是浮点数print(torch.linspace(start=torch.tensor(10), end=torch.tensor(20), steps=torch.tensor(4), dtype=torch.int64)) # 注意：结果仍然是浮点数print(torch.linspace(start=10.+6.j, end=20.+3.j, steps=4))print(torch.linspace(start=torch.tensor(10.+6.j), end=torch.tensor(20.+3.j), steps=torch.tensor(4)))print(torch.linspace(start=False, end=True, steps=4))print(torch.linspace(start=torch.tensor(True), end=torch.tensor(False), steps=torch.tensor(4)))

通过这些例子，您可以了解 linspace() 函数的用法以及其参数的意义。记住，对于整数类型输出，必须明确指定 dtype 参数。

文章推荐

PyTorch 中的 linspace

Python程序删除数组中的重复元素

使用Networxx模块的超链接诱导主题搜索（HITS）算法- Python

使用索引列表在Python中查找元素的乘积

在Python中打印字符串的所有子序列

使用Python从字符串的末尾删除给定的子字符串