pytorch中nonzero 的用法

nonzero（）：返回乙個包含輸入input中非0元素索引的張量（以張量的形式返回非0索引）

一維中：

import torch

a =torch.tensor([1
,2,4
,4,5
])print
(torch.nonzero(a<4)
)

print(torch.nonzero(a<4))：輸出布林值索引非0的位置

二維

import torch

python numpy中nonzero 的用法
當使用布林陣列直接作為下標物件或者元組下標物件中有布林陣列時，都相當於用nonzero 將布林陣列轉換成一組整數陣列，然後使用整數陣列進行下標運算。nonzeros a 返回陣列a中值不為零的元素的下標，它的返回值是乙個長度為a.ndim 陣列a的軸數 的元組，元組的每個元素都是乙個整數陣列，其值為...
python numpy中nonzero 的用法
當使用布林陣列直接作為下標物件或者元組下標物件中有布林陣列時，都相當於用nonzero 將布林陣列轉換成一組整數陣列，然後使用整數陣列進行下標運算。nonzeros a 返回陣列a中值不為零的元素的下標，它的返回值是乙個長度為a.ndim 陣列a的軸數 的元組，元組的每個元素都是乙個整數陣列，其值為...
python numpy中nonzero 的用法
當使用布林陣列直接作為下標物件或者元組下標物件中有布林陣列時，都相當於用nonzero 將布林陣列轉換成一組整數陣列，然後使用整數陣列進行下標運算。nonzeros a 返回陣列a中值不為零的元素的下標，它的返回值是乙個長度為a.ndim 陣列a的軸數 的元組，元組的每個元素都是乙個整數陣列，其值為...