PyTorch torch.ravel 関数

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

torch.ravelは、PyTorch でテンソルを1次元にフラット化するための関数です。可能であれば、入力テンソルのビューを返し、コピーは返しません。

関数定義

torch.ravel(input)

引数の説明：

  • input：入力テンソル

使用例

例

import torch

# 2x3 行列を作成
x = torch.tensor([[1, 2, 3], [4, 5, 6]])

# 1次元にフラット化する
y = torch.ravel(x)
print(y)

出力結果は：

tensor([1, 2, 3, 4, 5, 6])

例

import torch

# 3次元テンソルを作成
x = torch.arange(12).reshape(2, 3, 2)

# 1次元にフラット化する
y = torch.ravel(x)
print(y)

出力結果は：

tensor([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11])

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

その他の拡張