torch.相关函数

最新推荐文章于 2025-03-16 11:21:04 发布

Acowardintheworld

最新推荐文章于 2025-03-16 11:21:04 发布

阅读量413

点赞数

分类专栏： # PyTorch框架文章标签： pytorch python

本文链接：https://blog.youkuaiyun.com/cowardintheworld/article/details/120848687

版权

本文详细解读了PyTorch的nonzero()函数，展示了如何获取非零元素的索引，以及as_tuple参数的区别。重点介绍了默认行为和设置as_tuple为True时的输出形式。同时，介绍了flatten()函数用于重塑张量维度。

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

torch.nonzero()

Docstring:nonzero(input, *, out=None, as_tuple=False) -> LongTensor or tuple of LongTensors
涉及的参数主要是输入和 as_tuple，其实就是是否返回到一个元组，默认不是元组。
看下面的例子就懂了。

"""as_tuple=False的默认情况"""
>>> torch.nonzero(torch.tensor([1, 1, 1, 0, 1]))
    tensor([[ 0],
            [ 1],
            [ 2],
            [ 4]])     # 返回的是非0元素的索引，并且是索引的矩阵，shape= （非零元素,1）
            
>>> torch.nonzero(torch.tensor([[0.6, 0.0, 0.5, 0.0]