print(a.index_select(0, torch.tensor([0, 2]))) c = torch.index_select(a, 1, torch.tensor([1, 3])) print(c) 先定義了一個(gè)tensor,這里用到了linspace和view方法。 第一個(gè)參數(shù)是索引的對象,第二個(gè)參數(shù)0表示按行索引,1表示按列進(jìn)行索引,第三個(gè)參數(shù)是一個(gè)tensor,就是索引的序號,比如b里面tensor...
www.dbjr.com.cn/article/2036...htm 2025-6-5