Pytorch张量通过索引获取指定数据

2024-01-09 13:26:48

在这里插入图片描述

import torch
x = torch.tensor([1,2,3])

x
Out[3]: tensor([1, 2, 3])

x[0]  # 索引操作：取单个数字
Out[4]: tensor(1)

x[0:1]  # 切片操作：可以保持维度不变
Out[5]: tensor([1])

x[torch.tensor([True,False,True])]  # 布尔值索引，通过条件筛选，mask掩码
# 这个条件可以是tensor也可以是列表，只要长度一致
# x[[True,False,True]]
Out[6]: tensor([1, 3])

文章来源:https://blog.csdn.net/weixin_43941438/article/details/135475815
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。