pytorch的python API略读--tensor(三）

最新推荐文章于 2022-11-19 17:02:06 发布

机器视觉全栈er

最新推荐文章于 2022-11-19 17:02:06 发布

阅读量431

点赞数

分类专栏： pytorch的python API 文章标签： pytorch python 深度学习

本文链接：https://blog.youkuaiyun.com/qq_42322191/article/details/123122451

版权

本文介绍了PyTorch中针对Tensor的索引操作，包括使用torch.where根据布尔条件选择元素，torch.index_select沿指定维度筛选，以及torch.masked_select依据mask获取一维向量的方法。通过实例展示了这些函数的用法和效果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

作者：机器视觉全栈er
来源：cvtutorials.com

2.1.2 索引

筛选出符合某种条件的subtensor。

torch.where: 根据布尔变量的值选择tensor中的元素，用法如下：

torch.where(condition, x, y)

下面举个简单的例子：

>>> import torch
>>> cvtutorials = torch.randn(3, 4)
>>> threshold = torch.zeros(3, 4)
>>> cvtutorials
tensor([[-1.6981,  1.0443,  2.7922, -0.8736],
        [-2.0208, -0.4815, -0.1488, -0.9714],
        [ 1.1035,  0.4089,  0.6279,  2.4600]])
>>> torch.where(cvtutorials > 0, cvtutorials, threshold)
tensor(