tf.raw_ops.CollectiveBcastRecv
Получает значение тензора, рассылаемого с другого устройства.
tf.raw_ops.CollectiveBcastRecv(
T, group_size, group_key, instance_key, shape,
communication_hint='auto', timeout_seconds=0, name=None
)
| Аргументы | |
|---|---|
T | tf.DType из: tf.bool, tf.float32, tf.half, tf.float64, tf.int32, tf.int64. |
group_size | int. |
group_key | int. |
instance_key | int. |
shape | tf.TensorShape или список ints. |
communication_hint | Необязательный string. По умолчанию "auto". |
timeout_seconds | Необязательный float. По умолчанию 0. |
name | Название операции (необязательно). |
| Возвращаемое значение | |
|---|---|
Tensor типа T. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/CollectiveBcastRecv
