Spec-Zone.ru › TensorFlow 1.15

tf.keras.backend.cast

View source on GitHub

Преобразует тензор в другой тип данных и возвращает его.

Просмотр псевдонимов

Псевдонимы для миграции

См. Руководство по миграции для получения более подробной информации.

tf.compat.v1.keras.backend.cast, `tf.compat.v2.keras.backend.cast`

tf.keras.backend.cast(
    x, dtype
)

Вы можете преобразовать Keras переменную, но она по-прежнему вернёт Keras тензор.

Аргументы
x Keras тензор (или переменная).
dtype Строка, либо ('float16', 'float32', или 'float64').
Возвращает
Keras тензор с типом данных dtype.

Примеры:

Преобразовать переменную float32 в тензор float64

import tensorflow as tf
from tensorflow.keras import backend as K
input = K.ones(shape=(1,3))
print(input)
cast_input = K.cast(input, dtype='float64')
print(cast_input)

<tf.Variable 'Variable:0' shape=(1, 3) dtype=float32,
     numpy=array([[1., 1., 1.]], dtype=float32)>
tf.Tensor([[1. 1. 1.]], shape=(1, 3), dtype=float64)

© 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/keras/backend/cast

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API