tf.keras.distribution.distribute_tensor
Изменяет раскладку значения Tensor при выполнении функции jit.
tf.keras.distribution.distribute_tensor(
tensor, layout
)
| Аргументы | |
|---|---|
tensor | Tensor, раскладку которого нужно изменить. |
layout | TensorLayout, применяемая к значению. |
| Возвращает | |
|---|---|
| новое значение с указанной раскладкой тензора. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/distribution/distribute_tensor
