栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

keras 的dot函数

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

keras 的dot函数

首先先弄清楚哪是0轴(-1轴),1轴(-2轴)

看下面例子:

import tensorflow as tf

x = tf.constant([[1, 2, 3], [4, 5, 6]])
y2 = tf.reduce_sum(x, axis = 0)  
print("沿着张量0轴方向求和:n",y2.eval())
y3 = tf.reduce_sum(x, axis = 1)  
print("沿着张量1轴方向求和:n",y3.eval())

y4 = tf.reduce_sum(x, axis = 1, keep_dims=True) 
print("沿着张量1轴方向求和,并保持维度:n",y4.eval())

y5 = tf.reduce_sum(x, axis = -2)  
print("沿着张量-2轴方向求和:n",y5.eval())

y6 = tf.reduce_sum(x, axis = -1)  
print("沿着张量-1轴方向求和:n",y6.eval())

结果如下:

沿着张量0轴方向求和:

[5 7 9]

沿着张量1轴方向求和:

[ 6 15]

沿着张量1轴方向求和,并保持维度:

[[ 6] [15]]

沿着张量-2轴方向求和:

[5 7 9]

沿着张量1轴方向求和:

[ 6 15]

沿着张量-1轴方向求和:

[ 6 15]

 可以看到0轴为列,1轴为行,-1轴==1轴,-2轴==0轴

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/308226.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号