[tensorflow笔记]-tensorflow实现带mask的reduce_mean

本文介绍如何在TensorFlow中实现带mask的reduce_mean操作,避免在处理NLP任务时,padding补0部分影响平均值计算。通过基本操作,确保只对有效数据进行平均。
摘要由CSDN通过智能技术生成

在使用tensorflow处理一些tensor时,有时需要对一个tensor取平均,可以使用tf.reduce_mean操作,但是这个没法处理带有mask的tensor数据,本文主要就是利用tensorflow的基本操作实现带mask的平均。

tf.reduce_mean

比如我们的数据是3维tensor,shape=(B,N,H),B表示batch_size、N表示最大长度、H表示向量维度,这样的3维tensor在NLP相关模型中很常见。

# input tensor
tensor = tf.constant([[[1, 2, 3], [3, 4, 5], [5, 6, 7]], 
                      [[1, 1, 2], [3, 3, 4], [
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值