ubind按照不同维度切的理解

一直想不明白,在1维2维的是怎么且tensor的,画个图还好一点.
举个栗子:

a = torch.tensor([1,2,3,4,5,6,7,8])
b=a.view([2,2,2])
print(b)

c=torch.unbind(b,0)
print(c)

c=torch.unbind(b,1)
print(c)

c=torch.unbind(b,2)
print(c)

李姐

看下面图也可,其实就是,去掉一维度之后,就自动的从当前最低的维度开始,元素一个一个的吸附,组团,然后再到更高层次的一个维度去.