PyTorch torch.atleast_2d 関数
Pytorch torch リファレンスマニュアルtorch.atleast_2dは、PyTorch で入力テンソルを少なくとも 2 次元に変換するために使用される関数です。入力の次元が 2 次元未満の場合、次元が自動的に追加されて 2 次元になります。
関数定義
torch.atleast_2d(*tensors)
使用例
]) 形状: torch.Size([1, 1])
import torch
# スカラーを 2 次元テンソルに変換
x = torch.atleast_2d(5)
print("スカラー変換後:", x, "形状:", x.shape)
# 出力: スカラー変換後: tensor([
# 1 次元テンソルを 2 次元に変換
x = torch.tensor([1, 2, 3])
y = torch.atleast_2d(x)
print("1次元から2次元へ:", y, "形状:", y.shape)
# 出力: 1次元から2次元へ: tensor([[1, 2, 3]]) 形状: torch.Size([1, 3])
# すでに 2 次元のテンソルはそのまま保持される
x = torch.tensor([[1, 2], [3, 4]])
y = torch.atleast_2d(x)
print("2次元テンソル:", y, "形状:", y.shape)
# スカラーを 2 次元テンソルに変換
x = torch.atleast_2d(5)
print("スカラー変換後:", x, "形状:", x.shape)
# 出力: スカラー変換後: tensor([
# 1 次元テンソルを 2 次元に変換
x = torch.tensor([1, 2, 3])
y = torch.atleast_2d(x)
print("1次元から2次元へ:", y, "形状:", y.shape)
# 出力: 1次元から2次元へ: tensor([[1, 2, 3]]) 形状: torch.Size([1, 3])
# すでに 2 次元のテンソルはそのまま保持される
x = torch.tensor([[1, 2], [3, 4]])
y = torch.atleast_2d(x)
print("2次元テンソル:", y, "形状:", y.shape)
その他の拡張