Función PyTorch torch.unbind

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.unbindEs una función en PyTorch que se utiliza para dividir un tensor en una tupla a lo largo de la dimensión especificada.

Definición de la función

torch.unbind(input, dim=0)

Ejemplo de uso

Ejemplo

import torch

x = torch.arange(12).reshape(3, 4)
print(Tensor original:)
print(x)

result = torch.unbind(x, dim=0)
print(Dividir a lo largo de dim=0:)
for i, t in enumerate(result):
    print(fBloque {i}: {t})

result = torch.unbind(x, dim=1)
print(Dividir a lo largo de dim=1:)
for i, t in enumerate(result):
    print(fBloque {i}: {t})

El resultado de salida es:

原始张量:
tensor([[ 0,  1,  2,  3],
        [ 4,  5,  6,  7],
        [ 8,  9, 10, 11]])
沿 dim=0 分割:
  块 0: tensor([0, 1, 2, 3])
  Fragmento 1: tensor([4, 5, 6, 7])
  Fragmento 2: tensor([ 8,  9, 10, 11])
沿 dim=1 分割:
  块 0: tensor([0, 4, 8])
  Fragmento 1: tensor([1, 5, 9])
  Fragmento 2: tensor([ 2,  6, 10])
  Fragmento 3: tensor([ 3,  7, 11])

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones