tf.keras.ops.cast
Преобразование тензора к нужному типу.
tf.keras.ops.cast(
x, dtype
)
| Аргументы | |
|---|---|
x | Тензор или переменная. |
dtype | Целевой тип. |
| Возвращаемое значение | |
|---|---|
Тензор указанного dtype. |
Пример:
x = keras.ops.arange(4) x = keras.ops.cast(x, dtype="float16")
© 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/ops/cast
