torch数据的索引

最新推荐文章于 2024-07-25 09:18:49 发布

qianchedu

最新推荐文章于 2024-07-25 09:18:49 发布

阅读量966

点赞数

分类专栏： pytorch 文章标签： pytorch

本文链接：https://blog.csdn.net/qianchedu/article/details/123772541

版权

pytorch 专栏收录该内容

9 篇文章 0 订阅

订阅专栏

import torch

tensor = torch.arange(2,14)
print(tensor)
print(tensor[2])
print(tensor[1:4])
print(tensor[2:-1])
print(tensor[:5])
print(tensor[-3:])
index = [1,3,4,5,6]
print(tensor[index])

输出结果

tensor([ 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13])
tensor(4)
tensor([3, 4, 5])
tensor([ 4, 5, 6, 7, 8, 9, 10, 11, 12])
tensor([2, 3, 4, 5, 6])
tensor([11, 12, 13])
tensor([3, 5, 6, 7, 8])