Tensorflow——tf.cast()、tf.reduce_mean()

2023年9月18日 244次阅读 来源: SpareNoEfforts

tf.cast()

#强制类型转换
#将data_1(可以为int，bool，float,python列表，python元组
#【输出转换后的数据,原数据不变】）转换成指定dtype类型
tf.cast(data_1,dtype)

tf.reduce_mean()

#求input_data平均值
with tf.Session() as sess:
    a = [[1.,2.,3.,4.,5.],
         [2.,2.,4.,4.,4.]]
    s1 = tf.reduce_mean(a,0)#求第一维的平均值
    s = tf.reduce_mean(a,1)#求第二维的平均值
    print(sess.run(s))
    原文作者：SpareNoEfforts
    原文地址: https://www.jianshu.com/p/dc34124ecac3
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞