如何在不使用python索引的情况下切片torch张量

import torch

# 创建一个3x3的张量
x = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 在第1维度上切片，起始索引为0，切片长度为2
sliced_tensor = torch.narrow(x, 0, 0, 2)

print(sliced_tensor)

tensor([[1, 2, 3],
        [4, 5, 6]])

import torch

# 创建一个3x3的张量
x = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 创建一个索引张量，指定要切片的索引
indices = torch.tensor([0, 2])

# 在第0维度上根据索引切片
sliced_tensor = torch.index_select(x, 0, indices)

print(sliced_tensor)

tensor([[1, 2, 3],
        [7, 8, 9]])

import torch

# 创建一个3x3的张量
x = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 创建一个掩码张量，指定要选择的元素
mask = torch.tensor([[1, 0, 1], [0, 1, 0], [1, 0, 1]], dtype=torch.bool)

# 根据掩码选择元素
selected_elements = torch.masked_select(x, mask)

print(selected_elements)

tensor([1, 3, 5, 7, 9])