org.tensorflow.op.distribute

クラス

NcclAllReduce <T extends TNumber >すべての入力テンソルにわたるリダクションを含むテンソルを出力します。
NcclBroadcast <T extends TNumber >出力に接続されているすべてのデバイスに「入力」を送信します。
NcclReduce <T extends TNumber > 「reduction」を使用して「num_devices」から「input」を単一のデバイスに減らします。