Fait rouler les éléments d'un tenseur le long d'un axe.
Les éléments sont décalés positivement (vers des indices plus grands) par le décalage de « shift » le long de la dimension de « axis ». Les valeurs « shift » négatives déplaceront les éléments dans la direction opposée. Les éléments qui dépassent la dernière position reviendront à la première et vice versa. Plusieurs déplacements le long de plusieurs axes peuvent être spécifiés.
Par exemple:
# 't' is [0, 1, 2, 3, 4]
roll(t, shift=2, axis=0) ==> [3, 4, 0, 1, 2]
# shifting along multiple dimensions
# 't' is [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9]]
roll(t, shift=[1, -2], axis=[0, 1]) ==> [[7, 8, 9, 5, 6], [2, 3, 4, 0, 1]]
# shifting along the same axis multiple times
# 't' is [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9]]
roll(t, shift=[2, -3], axis=[1, 1]) ==> [[1, 2, 3, 4, 0], [6, 7, 8, 9, 5]]
Méthodes publiques
Sortie <T> | comme Sortie () Renvoie le handle symbolique d'un tenseur. |
statique <T, U étend le nombre, V étend le nombre> Roll <T> | |
Sortie <T> | sortir () A la même forme et la même taille que l'entrée. |
Méthodes héritées
Méthodes publiques
sortie publique <T> asOutput ()
Renvoie le handle symbolique d'un tenseur.
Les entrées des opérations TensorFlow sont les sorties d'une autre opération TensorFlow. Cette méthode est utilisée pour obtenir un handle symbolique qui représente le calcul de l’entrée.
public static Roll <T> créer (portée de portée , entrée d'opérande <T>, décalage d'opérande <U>, axe d'opérande <V>)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération Roll.
Paramètres
portée | portée actuelle |
---|---|
changement | La dimension doit être 0-D ou 1-D. `shift[i]` spécifie le nombre de places dont les éléments sont décalés positivement (vers des indices plus grands) le long de la dimension spécifiée par `axis[i]`. Les décalages négatifs feront rouler les éléments dans la direction opposée. |
axe | La dimension doit être 0-D ou 1-D. `axis[i]` spécifie la dimension dans laquelle le décalage `shift[i]` doit se produire. Si le même axe est référencé plusieurs fois, le décalage total pour cet axe sera la somme de tous les décalages appartenant à cet axe. |
Retours
- une nouvelle instance de Roll
sortie publique <T> sortie ()
A la même forme et la même taille que l'entrée. Les éléments sont décalés positivement (vers des indices plus grands) par les décalages de « shift » le long des dimensions de « axis ».