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
