在PyTorch中提取子张量

import torch

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

# 获取第一个元素
element = tensor[0, 0]  # 输出: tensor(1)

# 获取第二行的所有元素
row = tensor[1, :]  # 输出: tensor([4, 5, 6])

# 获取第三列的所有元素
column = tensor[:, 2]  # 输出: tensor([3, 6, 9])

# 切片出第一行和第二行
sub_tensor = tensor[0:2, :]  # 输出: tensor([[1, 2, 3],
                             #         [4, 5, 6]])

# 切片出第二列和第三列
sub_tensor = tensor[:, 1:3]  # 输出: tensor([[2, 3],
                             #         [5, 6],
                             #         [8, 9]])