Nota : Después de eliminar la dimensión especificada, se devuelve una tupla que contiene cada corte después de cortar a lo largo de la dimensión especificada.
Parámetro :
- tensor (Tensor) -tensor de entrada
- dimensión borrada (int)
>>> x = torch.randn(3, 3)
>>> x
tensor([[ 0.4775, 0.0161, -0.9403],
[ 1.6109, 2.1144, 1.1833],
[-0.2656, 0.7772, 0.5989]])
>>> torch.unbind(x, dim=1)(tensor([ 0.4775, 1.6109, -0.2656]), tensor([0.0161, 2.1144, 0.7772]), tensor([-0.9403, 1.1833, 0.5989]))
