tf.keras.distribution.distribute_tensor
Change the layout of a Tensor value in the jit function execution.
tf.keras.distribution.distribute_tensor(
tensor, layout
)
| Args | |
|---|---|
tensor | a Tensor to change the layout. |
layout | TensorLayout to be applied on the value. |
| Returns | |
|---|---|
| a new value with the specified 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
