Función torch.split de PyTorch
Manual de referencia de Pytorch torch
torch.splitEs una función de PyTorch utilizada para dividir un tensor a lo largo de una dimensión especificada.
Definición de la función
torch.split(tensor, split_size, dim)
Ejemplo de uso
Ejemplo
import torch
x = torch.arange(12).reshape(3, 4)
# Dividir a lo largo de 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)
# Dividir a lo largo de dim=0
result = torch.split(x, 1, dim=0)
for i, t in enumerate(result):
print(f"Part {i}:")
print(t)
Otras extensiones