美文网首页python与Tensorflow
Tensorflow——tf.cast()、tf.reduce_

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

作者: SpareNoEfforts | 来源:发表于2018-10-29 22:08 被阅读8次

    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))
    

    相关文章

      网友评论

        本文标题:Tensorflow——tf.cast()、tf.reduce_

        本文链接:https://www.haomeiwen.com/subject/dhiitqtx.html