tf.keras.distribution.distribute_tensor
Изменить структуру тензора в процессе выполнения функции jit.
tf.keras.distribution.distribute_tensor(
tensor, layout
)
| Аргументы | |
|---|---|
tensor | тензор, структуру которого необходимо изменить. |
layout | преобразование, которое будет применено к значению. |
| Возвращает | |
|---|---|
| новое значение с указанной структурой тензора. |
© 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
