PyTorch torch.full_like 関数
Pytorch torch リファレンスマニュアルtorch.full_likeこれは、PyTorch で入力テンソルと同じ形状の埋められたテンソルを作成する関数です。
関数定義
torch.full_like(input, fill_value, dtype=None, device=None, requires_grad=False)
使用例
実例
import torch
# 入力テンソルを作成
input = torch.tensor([[1, 2], [3, 4]])
# 入力と同じ形状の全要素 7 のテンソルを作成
x = torch.full_like(input, 7)
print(x)
# 入力テンソルを作成
input = torch.tensor([[1, 2], [3, 4]])
# 入力と同じ形状の全要素 7 のテンソルを作成
x = torch.full_like(input, 7)
print(x)
その他の拡張