tensorflow多维张量计算实例


Posted in Python onFebruary 11, 2020

两个三维矩阵的乘法怎样计算呢?我通过实验发现,tensorflow把前面的维度当成是batch,对最后两维进行普通的矩阵乘法。也就是说,最后两维之前的维度,都需要相同。

首先计算shape为(2, 2, 3)乘以shape为(2, 3, 2)的张量。

import tensorflow as tf
import numpy as np
a = tf.constant(np.arange(1, 13, dtype=np.float32), shape=[2, 2, 3])
b = tf.constant(np.arange(1, 13, dtype=np.float32), shape=[2, 3, 2])
c = tf.matmul(a, b)
# c = tf.matmul(a, b)
sess = tf.Session()
print("a*b = ", sess.run(c))
c1 = tf.matmul(a[0, :, :], b[0, :, :])
print("a[1]*b[1] = ", sess.run(c1))

运行结果:

tensorflow多维张量计算实例

计算结果表明,两个三维矩阵相乘,对应位置的最后两个维度的矩阵乘法。

再验证高维的张量乘法:

import tensorflow as tf
import numpy as np
a = tf.constant(np.arange(1, 36, dtype=np.float32), shape=[3, 2, 2, 3])
b = tf.constant(np.arange(1, 36, dtype=np.float32), shape=[3, 2, 3, 2])
c = tf.matmul(a, b)
# c = tf.matmul(a, b)
sess = tf.Session()
print("a*b = ", sess.run(c))
c1 = tf.matmul(a[0, 0, :, :], b[0, 0, :, :])
print("a[1]*b[1] = ", sess.run(c1))

运行结果:

tensorflow多维张量计算实例

以上这篇tensorflow多维张量计算实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Linux CentOS7下安装python3 的方法
Jan 21 Python
Python装饰器用法示例小结
Feb 11 Python
python sys,os,time模块的使用(包括时间格式的各种转换)
Apr 27 Python
Python实现的质因式分解算法示例
May 03 Python
python的内存管理和垃圾回收机制详解
May 18 Python
使用Pyhton集合set()实现成果查漏的例子
Nov 24 Python
python_mask_array的用法
Feb 18 Python
python topk()函数求最大和最小值实例
Apr 02 Python
spyder 在控制台(console)执行python文件,debug python程序方式
Apr 20 Python
15个应该掌握的Jupyter Notebook使用技巧(小结)
Sep 23 Python
python自动从arxiv下载paper的示例代码
Dec 05 Python
python中四舍五入的正确打开方式
Jan 18 Python
python误差棒图errorbar()函数实例解析
Feb 11 #Python
解决Python3.8用pip安装turtle-0.0.2出现错误问题
Feb 11 #Python
python scatter函数用法实例详解
Feb 11 #Python
python可视化text()函数使用详解
Feb 11 #Python
python读取图片的几种方式及图像宽和高的存储顺序
Feb 11 #Python
详解Python中的分支和循环结构
Feb 11 #Python
python re模块匹配贪婪和非贪婪模式详解
Feb 11 #Python
You might like
php 来访国内外IP判断代码并实现页面跳转
2009/12/18 PHP
php curl 登录163邮箱并抓取邮箱好友列表的代码(经测试)
2011/04/07 PHP
PHP_Cooikes不同页面无法传递的解决方法
2014/03/07 PHP
PHP输出XML格式数据的方法总结
2017/02/08 PHP
PHP检查端口是否可以被绑定的方法示例
2018/08/09 PHP
[JS]点出统计器
2020/10/11 Javascript
Javascript Object.extend
2010/05/18 Javascript
js验证模型自我实现的具体方法
2013/06/21 Javascript
Jquery实现图片左右自动滚动示例
2013/09/25 Javascript
input输入框鼠标焦点提示信息
2015/03/17 Javascript
jQuery实现鼠标悬停显示提示信息窗口的方法
2015/04/30 Javascript
AngularJS实现元素显示和隐藏的几个案例
2015/12/09 Javascript
js实现select二级联动下拉菜单
2020/04/17 Javascript
JS加载器如何动态加载外部js文件
2016/05/26 Javascript
Javascript缓存API
2016/06/14 Javascript
原生js获取浏览器窗口及元素宽高常用方法集合
2017/01/18 Javascript
vue jsx 使用指南及vue.js 使用jsx语法的方法
2017/11/11 Javascript
利用Blob进行文件上传的完整步骤
2018/08/02 Javascript
解决vue keep-alive 数据更新的问题
2018/09/21 Javascript
使用apifm-wxapi模块中的问题及解决方法
2019/08/05 Javascript
js实现九宫格布局效果
2020/05/28 Javascript
element-ui中dialog弹窗关闭按钮失效的解决
2020/09/22 Javascript
Vue3 实现双盒子定位Overlay的示例
2020/12/22 Vue.js
Vue ​v-model相关知识总结
2021/01/28 Vue.js
[01:11:02]Secret vs Newbee 2019国际邀请赛小组赛 BO2 第一场 8.15
2019/08/17 DOTA
python显示天气预报
2014/03/02 Python
Python3实现将本地JSON大数据文件写入MySQL数据库的方法
2018/06/13 Python
Python json模块与jsonpath模块区别详解
2020/03/05 Python
Pytho爬虫中Requests设置请求头Headers的方法
2020/09/22 Python
英国复古服装购物网站:Collectif
2019/10/30 全球购物
中餐厅经理岗位职责
2014/04/11 职场文书
如何写贫困证明申请书
2014/10/29 职场文书
2014年小学体育工作总结
2014/12/11 职场文书
历史名人教你十五个读书方法,赶快Get起来!
2019/07/18 职场文书
python图片灰度化处理的几种方法
2021/06/23 Python
springboot创建的web项目整合Quartz框架的项目实践
2022/06/21 Java/Android