tf.cast()
Posted happytaiyang
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了tf.cast()相关的知识,希望对你有一定的参考价值。
tf.cast()函数的作用是执行 tensorflow 中张量数据类型转换,比如读入的图片如果是uint8类型的,一般在要在训练前把图像的数据格式转换为float32。
cast定义:
cast(x, dtype, name=None)
第一个参数 x: 待转换的数据(张量)
第二个参数 dtype: 目标数据类型
第三个参数 name: 可选参数,定义操作的名称
以上是关于tf.cast()的主要内容,如果未能解决你的问题,请参考以下文章
TFboy养成记 tf.cast,tf.argmax,tf.reduce_sum
tensorrt不支持:tf.unpack,tf.slice,tf.tile,tf.expand_dims,tf.fill,tf.cast,tf.floor_div,tf.range