PyTorch torch.empty_strided 関数
Pytorch torch リファレンスマニュアルtorch.empty_stridedPyTorch において、指定されたストライドを持つ未初期化テンソルを作成するための関数です。
関数定義
torch.empty_strided(size, stride, dtype=None, device=None, requires_grad=False)
使用例
実例
import torch
# 形状が (3, 4)、ストライドが (1, 3) のテンソルを作成
x = torch.empty_strided((3, 4), (1, 3))
print(x)
print("形状:", x.shape)
print("ストライド:", x.stride())
print("注意: 値は未初期化のメモリ値です")
# 形状が (3, 4)、ストライドが (1, 3) のテンソルを作成
x = torch.empty_strided((3, 4), (1, 3))
print(x)
print("形状:", x.shape)
print("ストライド:", x.stride())
print("注意: 値は未初期化のメモリ値です")
その他の拡張機能