Questions et réponses pour le tensor-indexing :

Questions liées à l'indexation et à l'indexation de tableaux et tenseurs multidimensionnels, dans un esprit similaire à l'indexation matricielle des balises

1
votes

Pour une condition donnée, obtenir des indices de valeurs dans le tenseur 2D A, utiliser ceux-ci pour indexer un tenseur 3D B

Pour un tenseur 2D donné, je veux récupérer tous les indices dont la valeur est 1 . Je m'attendais à pouvoir utiliser simplement torch.nonzero (a = ...

python multidimensional-array pytorch tensor tensor-indexing

0 commentaire(s)