Avertissement : Cette API est obsolète et sera supprimée dans une future version de TensorFlow une fois que le remplacement sera stable.
NcclAllReduce
Restez organisé à l'aide des collections
Enregistrez et classez les contenus selon vos préférences.
Génère un tenseur contenant la réduction sur tous les tenseurs d'entrée.
Génère un tenseur contenant la réduction sur tous les tenseurs d'entrée transmis aux opérations dans le même « nom_partagé ».
Le graphique doit être construit de sorte que si une opération s'exécute avec la valeur de nom_partagé `c`, alors les opérations `num_devices` s'exécuteront avec la valeur de nom_partagé `c`. Si vous ne le faites pas, l’exécution du graphique ne pourra pas se terminer.
input : l'entrée des données de réduction : la valeur de la réduction sur tous les appareils `num_devices`. réduction : l’opération de réduction à effectuer. num_devices : le nombre d'appareils participant à cette réduction. shared_name : identifiant partagé entre les opérations de la même réduction.
Méthodes publiques
Sortie <T> | |
statique <T étend le nombre> NcclAllReduce <T> | créer (portée de portée , entrée opérande <T>, réduction de chaîne, numDevices longs, chaîne sharedName) Méthode d'usine pour créer une classe encapsulant une nouvelle opération NcclAllReduce. |
Sortie <T> | |
Méthodes héritées
De la classe java.lang.Object booléen | est égal à (Objet arg0) |
Classe finale<?> | obtenirClasse () |
int | Code de hachage () |
vide final | notifier () |
vide final | notifierTous () |
Chaîne | àChaîne () |
vide final | attendre (long arg0, int arg1) |
vide final | attendez (long arg0) |
vide final | attendez () |
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.
Méthode d'usine pour créer une classe encapsulant une nouvelle opération NcclAllReduce.
Retour
- une nouvelle instance de NcclAllReduce
Sauf indication contraire, le contenu de cette page est régi par une licence Creative Commons Attribution 4.0, et les échantillons de code sont régis par une licence Apache 2.0. Pour en savoir plus, consultez les Règles du site Google Developers. Java est une marque déposée d'Oracle et/ou de ses sociétés affiliées.
Dernière mise à jour le 2023/12/01 (UTC).
[{
"type": "thumb-down",
"id": "missingTheInformationINeed",
"label":"Il n'y a pas l'information dont j'ai besoin"
},{
"type": "thumb-down",
"id": "tooComplicatedTooManySteps",
"label":"Trop compliqué/Trop d'étapes"
},{
"type": "thumb-down",
"id": "outOfDate",
"label":"Obsolète"
},{
"type": "thumb-down",
"id": "translationIssue",
"label":"Problème de traduction"
},{
"type": "thumb-down",
"id": "samplesCodeIssue",
"label":"Mauvais exemple/Erreur de code"
},{
"type": "thumb-down",
"id": "otherDown",
"label":"Autre"
}]
[{
"type": "thumb-up",
"id": "easyToUnderstand",
"label":"Facile à comprendre"
},{
"type": "thumb-up",
"id": "solvedMyProblem",
"label":"J'ai pu résoudre mon problème"
},{
"type": "thumb-up",
"id": "otherUp",
"label":"Autre"
}]
{"lastModified": "Derni\u00e8re mise \u00e0 jour le 2023/12/01\u00a0(UTC)."}