tf.cast
| Просмотр исходного кода на GitHub |
Преобразует тензор в новый тип.
tf.cast(
x, dtype, name=None
)
Операция преобразует x (в случае Tensor) или x.values (в случае SparseTensor или IndexedSlices) в dtype.
Например:
x = tf.constant([1.8, 2.2], dtype=tf.float32) tf.dtypes.cast(x, tf.int32) # [1, 2], dtype=tf.int32
Операция поддерживает типы данных (для x и dtype) uint8, uint16, uint32, uint64, int8, int16, int32, int64, float16, float32, float64, complex64, complex128, bfloat16. В случае преобразования из комплексных типов (complex64, complex128) в вещественные типы возвращается только вещественная часть x. В случае преобразования из вещественных типов в комплексные типы (complex64, complex128) мнимая часть возвращаемого значения устанавливается в 0. Обработка комплексных типов здесь соответствует поведению numpy.
| Аргументы | |
|---|---|
x | Tensor или SparseTensor или IndexedSlices числового типа. Может быть uint8, uint16, uint32, uint64, int8, int16, int32, int64, float16, float32, float64, complex64, complex128, bfloat16 . |
dtype | Целевой тип. Список поддерживаемых типов данных такой же, как у x. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
Tensor или SparseTensor или IndexedSlices с такой же формой, как у x и таким же типом, как у dtype. |
| Исключения | |
|---|---|
TypeError | Если x нельзя преобразовать в dtype. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/cast
