tf.keras.distribution.TensorLayout
Макет для применения к тензору.
tf.keras.distribution.TensorLayout(
axes, device_mesh=None
)
Этот API согласован с jax.sharding.NamedSharding и tf.dtensor.Layout.
Дополнительные сведения см. в jax.sharding.NamedSharding и tf.dtensor.Layout.
| Аргументы | |
|---|---|
axes | кортеж строк, которые должны соответствовать axis_names в DeviceMesh. Для любых измерений, для которых не требуется фрагментация, можно использовать None в качестве заглушки. |
device_mesh | Необязательная DeviceMesh, которая будет использована для создания макета. Фактическое сопоставление тензора с физическим устройством неизвестно до тех пор, пока не будет указана сетка. |
| Атрибуты | |
|---|---|
axes | |
device_mesh | |
© 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/TensorLayout
