PyTorch torch.split 関数
Pytorch torch リファレンスマニュアルtorch.splitPyTorch において、テンソルを指定した次元に沿って分割するための関数です。
関数定義
torch.split(tensor, split_size, dim)
使用例
実例
import torch
x = torch.arange(12).reshape(3, 4)
# dim=0 に沿って分割
result = torch.split(x, 1, dim=0)
for i, t in enumerate(result):
print(f"Part {i}:")
print(t)
x = torch.arange(12).reshape(3, 4)
# dim=0 に沿って分割
result = torch.split(x, 1, dim=0)
for i, t in enumerate(result):
print(f"Part {i}:")
print(t)
その他の拡張