x = torch.tensor([[1,3,5,7,9],[2,4,6,8,10]])
print(x[0:2])
# 表示从第一维度(行)开始输出第0行第一行
# 输出结果如下
# tensor([[ 1, 3, 5, 7, 9],
# [ 2, 4, 6, 8, 10]])
print(x[...,0::2])
# 表示从第二维度(列)开始以步长为2的距离开始输出
# 输出结果如下
# tensor([[ 1, 5, 9],
# [ 2, 6, 10]])
python二维列表的切片
最新推荐文章于 2024-07-20 17:12:48 发布
文章展示了如何使用PyTorch中的tensor对象进行索引和切片,包括按行选取及按列以特定步长选取数据,输出结果为简化后的二维数组。
摘要由CSDN通过智能技术生成