PyTorch torch.atleast_1d 関数
Pytorch torch リファレンスマニュアルtorch.atleast_1dPyTorch において、入力テンソルを少なくとも1次元に変換するための関数です。入力がすでに少なくとも1次元である場合は、元のテンソルのビューを返します。それ以外の場合は、新しい次元が追加されます。
関数定義
torch.atleast_1d(*tensors)
使用例
例
import torch
# スカラーを1次元テンソルに変換する
x = torch.atleast_1d(5)
print("スカラー変換後:", x, "形状:", x.shape)
# 既に1次元のテンソル
x = torch.tensor([1, 2, 3])
y = torch.atleast_1d(x)
print("1次元テンソル:", y, "形状:", y.shape)
# 2次元テンソルはそのまま
x = torch.tensor([[1, 2], [3, 4]])
y = torch.atleast_1d(x)
print("2次元テンソル:", y, "形状:", y.shape)
# 複数の入力
a = 1
b = torch.tensor([2, 3])
c = torch.tensor([[4, 5]])
result = torch.atleast_1d(a, b, c)
for i, t in enumerate(result):
print(f"入力{i}の形状: {t.shape}")
# スカラーを1次元テンソルに変換する
x = torch.atleast_1d(5)
print("スカラー変換後:", x, "形状:", x.shape)
# 既に1次元のテンソル
x = torch.tensor([1, 2, 3])
y = torch.atleast_1d(x)
print("1次元テンソル:", y, "形状:", y.shape)
# 2次元テンソルはそのまま
x = torch.tensor([[1, 2], [3, 4]])
y = torch.atleast_1d(x)
print("2次元テンソル:", y, "形状:", y.shape)
# 複数の入力
a = 1
b = torch.tensor([2, 3])
c = torch.tensor([[4, 5]])
result = torch.atleast_1d(a, b, c)
for i, t in enumerate(result):
print(f"入力{i}の形状: {t.shape}")
その他の拡張