tensor 操作學(xué)習(xí)

Notice: torch.split() 第二個(gè)參數(shù)為list 的時(shí)候，list內(nèi)元素之和要等于指定維度的值

Notice: index的 dtype=torch.long 64位整型

Notice: 只有軸的長度為1時(shí)才會(huì)被移除??！