PyTorch torch.empty_like 関数
torch.empty_likeは PyTorch において、入力テンソルと同じ形状の未初期化テンソルを作成するための関数です。
関数定義
torch.empty_like(input, dtype=None, device=None, requires_grad=False)
使用例
実例
import torch
# 入力テンソルを作成する
input = torch.tensor([[1, 2], [3, 4]])
# 入力と同じ形状の未初期化テンソルを作成する
x = torch.empty_like(input)
print(x)
print(注意: 値は未初期化のメモリ値です)
# 入力テンソルを作成する
input = torch.tensor([[1, 2], [3, 4]])
# 入力と同じ形状の未初期化テンソルを作成する
x = torch.empty_like(input)
print(x)
print(注意: 値は未初期化のメモリ値です)
その他の拡張
Pytorch torch リファレンスマニュアル