tensorflow中 tf.equal、tf.cast、tf.reduce_mean函数使用

本文详细介绍了TensorFlow中三个常用函数的使用方法:tf.equal()用于比较两个张量的元素是否相等,tf.cast()用于转换张量的数据类型,tf.reduce_mean()用于计算张量的平均值。通过具体示例展示了这些函数如何在实际场景中应用。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

一、tf.equal()

      使用方法:tf.equal(a,b)

      判断 a 和 b 对应位置的值是否相等,相等则返回 true , 不等则返回 false   (注意返回的是布尔值)

二、tf.cast()

      使用方法:tf.cast(a,dtype)         

      将输入的 a 的数据类型转成指定的数据类型(dtype)。

三、tf.reduce_mean()

       使用方法:tf.reduce_mean(a)   

       求a的平均值,返回是一个标量,也就是一个数字,不论a是多少维的

四、下面看一个小例子

 

import tensorflow as tf
a = [[1,2,2],[1,3,4]]
b = [[1,2,3],[3,5,6]]
c = tf.equal(a,b)
d = tf.cast(c,dtype=tf.float32)
e = tf.reduce_mean(d)
with tf.Session() as sess:
    c,d,e = sess.run([c,d,e])
    print(c)
    print(d)
    print(e)

 


输出结果:

 


[[ True  True False]
 [False False False]]
[[1. 1. 0.]
 [0. 0. 0.]]
0.33333334


注意:equal和cast都不会改变输入的维度,reduce_mean输出一个数值

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值