PyTorch torch.sparse_csr_tensor 関数
Pytorch torch リファレンスマニュアルtorch.sparse_csr_tensorPyTorch でスパース CSR（圧縮スパース行）テンソルを構築するための関数です。
関数定義
torch.sparse_csr_tensor(crow_indices, col_indices, values, size=None, dtype=None, device=None, requires_grad=False)
使用例
例
import torch
# スパース CSR テンソルを作成
# 行ポインタ
crow_indices = torch.tensor([0, 2, 3])
# 列インデックス
col_indices = torch.tensor([0, 1, 1])
# 値
values = torch.tensor([1.0, 2.0, 3.0])
# テンソルのサイズ
size = (2, 3)
# スパース CSR テンソルを構築
x = torch.sparse_csr_tensor(crow_indices, col_indices, values, size)
print(x)
print(密なテンソル:)
print(x.to_dense())
# スパース CSR テンソルを作成
# 行ポインタ
crow_indices = torch.tensor([0, 2, 3])
# 列インデックス
col_indices = torch.tensor([0, 1, 1])
# 値
values = torch.tensor([1.0, 2.0, 3.0])
# テンソルのサイズ
size = (2, 3)
# スパース CSR テンソルを構築
x = torch.sparse_csr_tensor(crow_indices, col_indices, values, size)
print(x)
print(密なテンソル:)
print(x.to_dense())
その他の拡張