import torch
A_idx = torch.LongTensor([0, 2]) # the index vector
B = torch.LongTensor([[1, 2, 3], [4, 5, 6]])
C = B.index_select(1, A_idx)
# 1 3
# 4 6
pytorch的tf.slice
阅读 67
2022-07-27
相关推荐
精彩评论（0）
0 0 举报
pytorch的tf.slice
阅读 67
2022-07-27
import torch
A_idx = torch.LongTensor([0, 2]) # the index vector
B = torch.LongTensor([[1, 2, 3], [4, 5, 6]])
C = B.index_select(1, A_idx)
# 1 3
# 4 6
相关推荐
精彩评论（0）