cast¶
dragon.vm.tensorflow.
cast
(
x,
dtype,
name=None
)[source]¶Cast the data type of input.
Examples:
x = tf.constant([1, 2, 3], dtype='int64') print(tf.cast(x, dtype='float32'))
- Parameters:
- x (dragon.Tensor) – The input tensor.
- dtype (str) – The data type to cast to.
- name (str, optional) – The operation name.
- Returns:
dragon.Tensor – The output tensor.