• 欢迎访问搞代码网站,推荐使用最新版火狐浏览器和Chrome浏览器访问本网站!
  • 如果您觉得本站非常有看点,那么赶紧使用Ctrl+D 收藏搞代码吧

对tf.reduce_sum tensorflow维度上的操作详解

python 搞代码 4年前 (2022-01-07) 28次浏览 已收录 0个评论

今天小编就为大家分享一篇对tf.reduce_sum tensorflow维度上的操作详解,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧

tensorflow中有很多在维度上的操作,本例以常用的tf.reduce_sum进行说明。官方给的api

 reduce_sum( input_tensor, axis=None, keep_dims=False, name=None, reduction_indices=None )

input_tensor:表示输入

axis:表示在那个维度进行sum操作。

keep_dims:表示是否保留原始数据的维度,False相当于执行完后原始数据就会少一个维度。

reduction_indices:为了跟旧版本的兼容,现在已经不使用了。

官方的例子:

 # 'x' is [[1, 1, 1] #   [1, 1, 1]] tf.reduce_sum(x) ==> 6 tf.reduce_sum(x, 0) ==> [2, 2, 2] tf.reduce_sum(x, 1) ==> [3, 3] tf.reduce_sum(x, 1, keep_dims=True) ==> [[3], [3]] tf.reduce_sum(x, [0, 1]) ==> 6

自己做的例子:

 import tensorflow as tf import numpy as np x = np.asarray([[[1,2,3],[4,5,6]],[[7,8,9],[10,11,12]]]) x_p = tf.placeholder(tf.int3<span style="color:transparent">来源gaodai#ma#com搞*代#码网</span>2,[2,2,3]) y = tf.reduce_sum(x_p,0) #修改这里 with tf.Session() as sess: y = sess.run(y,feed_dict={x_p:x}) print y
 axis= 0:[[ 8 10 12] [14 16 18]] 1+7 2+8 3+7 …….. axis=1: [[ 5 7 9] [17 19 21]] 1+4 2+5 3 +6 …. axis=2: [[ 6 15] [24 33]] 1+2+3 4+5+6…..

以上这篇对tf.reduce_sum tensorflow维度上的操作详解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持gaodaima搞代码网

以上就是对tf.reduce_sum tensorflow维度上的操作详解的详细内容,更多请关注gaodaima搞代码网其它相关文章!


搞代码网(gaodaima.com)提供的所有资源部分来自互联网,如果有侵犯您的版权或其他权益,请说明详细缘由并提供版权或权益证明然后发送到邮箱[email protected],我们会在看到邮件的第一时间内为您处理,或直接联系QQ:872152909。本网站采用BY-NC-SA协议进行授权
转载请注明原文链接:对tf.reduce_sum tensorflow维度上的操作详解

喜欢 (0)
[搞代码]
分享 (0)
发表我的评论
取消评论

表情 贴图 加粗 删除线 居中 斜体 签到

Hi,您需要填写昵称和邮箱!

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址