Pytorch tensor特定條件判斷索引

torch.where() 用於將兩個broadcastable的tensor組合成新的tensor，類似於c++中的三元操作符「?:」 區別於python numpy中的where()直接可以找到特定條件元素的index 想要實現numpy中where()的功能，可以藉助nonzero() 對應numpy中的where()操作效果:

>>阅读原文<<