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 相关文章推荐
python随机生成指定长度密码的方法
Apr 04 Python
简单理解Python中的装饰器
Jul 31 Python
Python中防止sql注入的方法详解
Feb 25 Python
使用python实现http及ftp服务进行数据传输的方法
Oct 26 Python
Python实例方法、类方法、静态方法的区别与作用详解
Mar 25 Python
python tkinter实现界面切换的示例代码
Jun 14 Python
Django如何自定义model创建数据库索引的顺序
Jun 20 Python
详解python实现小波变换的一个简单例子
Jul 18 Python
Python *args和**kwargs用法实例解析
Mar 02 Python
Python根据字典的值查询出对应的键的方法
Sep 30 Python
Cpython解释器中的GIL全局解释器锁
Nov 09 Python
OpenCV绘制圆端矩形的示例代码
Aug 30 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 SQLite类
2009/05/07 PHP
PHP学习笔记之一
2011/01/17 PHP
php通过隐藏表单控件获取到前两个页面的url
2014/09/09 PHP
php的4种常见运行方式
2015/03/20 PHP
php生成动态验证码gif图片
2015/10/19 PHP
YII分模块加载路由的实现方法
2018/10/01 PHP
jQuery DIV弹出效果实现代码
2009/07/03 Javascript
jQuery 获取兄弟元素的几种不错方法
2014/05/23 Javascript
jQuery页面加载初始化常用的三种方法
2014/06/04 Javascript
jQuery+jRange实现滑动选取数值范围特效
2015/03/14 Javascript
百度地图API之本地搜索与范围搜索
2015/07/30 Javascript
jQuery点击按钮弹出遮罩层且内容居中特效
2015/12/14 Javascript
一句jQuery代码实现返回顶部效果(简单实用)
2016/12/28 Javascript
jQuery插件Echarts实现的渐变色柱状图
2017/03/23 jQuery
select获取下拉框的值 下拉框默认选中方法
2018/02/28 Javascript
详解IOS微信上Vue单页面应用JSSDK签名失败解决方案
2018/11/14 Javascript
Electron-vue开发的客户端支付收款工具的实现
2019/05/24 Javascript
javascript实现自由编辑图片代码详解
2019/06/21 Javascript
python遍历文件夹,指定遍历深度与忽略目录的方法
2018/07/11 Python
python中文编码与json中文输出问题详解
2018/08/24 Python
Python异常处理知识点总结
2019/02/18 Python
python读取ini配置的类封装代码实例
2020/01/08 Python
基于Keras 循环训练模型跑数据时内存泄漏的解决方式
2020/06/11 Python
python利用google翻译方法实例(翻译字幕文件)
2020/09/21 Python
Python创建简单的神经网络实例讲解
2021/01/04 Python
Python实现我的世界小游戏源代码
2021/03/02 Python
英国排名第一的宠物店:PetPlanet
2020/02/02 全球购物
函授自我鉴定范文
2014/02/06 职场文书
2014年国庆标语
2014/06/30 职场文书
英语感谢信范文
2015/01/20 职场文书
世界遗产的导游词
2015/02/13 职场文书
2015年乡镇扶贫工作总结
2015/04/08 职场文书
2016年七夕爱情寄语
2015/12/04 职场文书
如何用PHP实现多线程编程
2021/05/26 PHP
Python数据可视化之基于pyecharts实现的地理图表的绘制
2021/06/10 Python
解决mysql问题:由于找不到MSVCR120.dll,无法继续执行代码
2021/06/26 MySQL