tf.reduce_mean(tf.cast(tf.equal(tf.argmax(out, 1), tf.argmax(y_, 1)), tf.float32))解析

1.tf.argmax()

返回 arr 的最大值
with tf.Session() as sess:
    print(sess.run(tf.argmax(arr, 0)))# 返回每一列的最大值的索引
    print(sess.run(tf.argmax(arr, 1)))# 返回每一行的最大值的索引

2.tf.equal()

比较data_1和data_2是否一样,
一样返回True
否则返回False

3.tf.cast()

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

4.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))
注意:结果的类型与你的数据内容的类型一致
  • 2
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值