Función torch.chunk de PyTorch
Manual de referencia de Pytorch torch
torch.chunkEs una función de PyTorch que se utiliza para dividir un tensor en trozos a lo largo de una dimensión especificada.
Definición de la función
torch.chunk(tensor, chunks, dim)
Ejemplo de uso
Ejemplo
import torch
x = torch.arange(12).reshape(3, 4)
# Dividir en 3 trozos
result = torch.chunk(x, 3, dim=0)
print("Número de trozos:", len(result))
for i, t in enumerate(result):
print(f"Chunk {i}:", t.shape)
x = torch.arange(12).reshape(3, 4)
# Dividir en 3 trozos
result = torch.chunk(x, 3, dim=0)
print("Número de trozos:", len(result))
for i, t in enumerate(result):
print(f"Chunk {i}:", t.shape)
Otras extensiones