PyTorch torch.ones 関数

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

torch.onesは PyTorch で全1テンソルを作成するための関数です。指定された形状のテンソルを作成し、すべての要素が1に初期化されます。

これは深層学習において、特定のパラメータの初期化、マスクの作成、または数学演算の基準値としてよく使用されます。

関数定義

torch.ones(*size, dtype=None, device=None, requires_grad=False, pin_memory=False)

引数:

  • *size(int): テンソルの形状。例:3、(3, 4)、(2, 3, 4)など。
  • dtype(torch.dtype, オプション): テンソルのデータ型を指定します。デフォルトはtorch.float32。
  • device(torch.device, オプション): テンソルを格納するデバイスを指定します。
  • requires_grad(bool, オプション): 勾配の計算が必要かどうか。
  • pin_memory(bool, オプション): ページロックメモリを使用するかどうか。

戻り値:

  • torch.Tensor: 全1テンソルを返します。

使用例

例1: 1次元の全1テンソルを作成

実例

import torch

# 5つの要素を含む全1テンソルを作成
x = torch.ones(5)

print(x)

出力結果は次のとおりです:

tensor([1., 1., 1., 1., 1.])

例2: 2次元の全1テンソルを作成

実例

import torch

# 3x4の全1テンソル（行列）を作成
x = torch.ones(3, 4)

print(x)
print(x.shape)

出力結果は次のとおりです:

tensor([[1., 1., 1., 1.],
        [1., 1., 1., 1.],
        [1., 1., 1., 1.]])
torch.Size([3, 4])

例3: 3次元の全1テンソルを作成

実例

import torch

# 2x3x4の全1テンソルを作成
x = torch.ones(2, 3, 4)

print(x.shape)

出力結果は次のとおりです:

torch.Size([2, 3, 4])

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

その他の拡張