PyTorch torch.range 関数
Pytorch torch リファレンスマニュアルtorch.rangePyTorch において end の値を含む等差数列テンソルを作成するための関数です。
関数定義
torch.range(start=0, end, step=1, dtype=None, device=None, requires_grad=False)
使用例
例
import torch
# 0から5までの等差数列を作成（5を含む）
x = torch.range(5)
print(x)
# 1から10まで、ステップ2の等差数列を作成
y = torch.range(1, 10, 2)
print(y)
# 0から5までの等差数列を作成（5を含む）
x = torch.range(5)
print(x)
# 1から10まで、ステップ2の等差数列を作成
y = torch.range(1, 10, 2)
print(y)
その他の拡張