TensorFlow类型转换函数,其函数原型是
时间: 2024-05-03 08:19:41 浏览: 88
在 TensorFlow 中,常用的类型转换函数是tf.cast(),其函数原型如下:
```
tf.cast(x, dtype, name=None)
```
其中,参数含义如下:
- x:待转换的张量。
- dtype:目标类型。
- name:操作的名称(可选)。
例如,将一个 float32 类型的张量转换为 int32 类型,可以使用如下代码:
```
import tensorflow as tf
x = tf.constant([1.2, 2.3, 3.4], dtype=tf.float32)
y = tf.cast(x, dtype=tf.int32)
print(y)
```
输出结果为:
```
tf.Tensor([1 2 3], shape=(3,), dtype=int32)
```
阅读全文