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