PyTorch torch.arange 関数

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

torch.arangePyTorch で等差数列のテンソルを作成するための関数です。開始値から終了値までの等差数列を含む一次元テンソルを作成します。

これは深層学習において、インデックスの作成、範囲配列の作成、またはループ内でのシーケンス生成によく使用されます。

関数定義

torch.arange(start=0, end, step=1, dtype=None, device=None, requires_grad=False)

引数:

  • start(float, オプション): シーケンスの開始値。デフォルトは0。
  • end(float): シーケンスの終了値（含まない）。
  • step(float, オプション): ステップ幅。デフォルトは1。
  • dtype(torch.dtype, オプション): テンソルのデータ型を指定します。
  • device(torch.device, オプション): テンソルを保存するデバイスを指定します。
  • requires_grad(bool, オプション): 勾配を計算する必要があるかどうか。

戻り値:

  • torch.Tensor: 一次元テンソルを返します。

使用例

例1: 0から5まで

例

import torch

# 0から4までの等差数列を作成
x = torch.arange(5)

print(x)

出力結果：

tensor([0, 1, 2, 3, 4])

例2: 開始値と終了値を指定

例

import torch

# 2から8までの等差数列を作成
x = torch.arange(2, 9)

print(x)

出力結果：

tensor([2, 3, 4, 5, 6, 7, 8])

例3: ステップ幅を指定

例

import torch

# 0から10まで、ステップ幅2の等差数列を作成
x = torch.arange(0, 11, 2)

print(x)

出力結果：

tensor([0, 2, 4, 6, 8, 10])

例4: 負のステップ幅

例

import torch

# 10から0まで、ステップ幅-2の等差数列を作成
x = torch.arange(10, 0, -2)

print(x)

出力結果：

tensor([10,  8,  6,  4,  2])

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

その他の拡張