景墨轩

学习pytorch中，看到文档里关于torch.nonzero的介绍和举例，一维的那个例子还好理解，二维的就不是很理解了，不明白为什么会出现两个[0,1,2,3]，于是再网上查看了一些讲解，终于搞明白了是怎么回事。先看一下原函数中每个元素的介绍torch.nonzero(input, *, out=None, as_tuple=False)input：输入的必须是tensorout：输出 Z × N ， N 代表输入数据的维度， Z 是总共非0元素的个数as_tuple：if as_tuple