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实现简单拆分PDF文件的方法
Jul 30 Python
python递归打印某个目录的内容(实例讲解)
Aug 30 Python
python3设计模式之简单工厂模式
Oct 17 Python
python机器学习之随机森林(七)
Mar 26 Python
在Python中增加和插入元素的示例
Nov 01 Python
pycharm新建一个python工程步骤
Jul 16 Python
TensorFlow实现指数衰减学习率的方法
Feb 05 Python
Python通过Pillow实现图片对比
Apr 29 Python
TensorFlow的环境配置与安装教程详解(win10+GeForce GTX1060+CUDA 9.0+cuDNN7.3+tensorflow-gpu 1.12.0+python3.5.5)
Jun 22 Python
Python实现一个简单的递归下降分析器
Aug 01 Python
python 使用建议与技巧分享(四)
Aug 18 Python
python爬虫线程池案例详解(梨视频短视频爬取)
Feb 20 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+AJAX实现无刷新注册(带用户名实时检测)
2006/12/02 PHP
php预定义变量使用帮助(带实例)
2013/10/30 PHP
php+mysql实现无限分类实例详解
2015/01/15 PHP
php提高网站效率的技巧
2015/09/29 PHP
Yii框架数据库查询、增加、删除操作示例
2019/10/14 PHP
javascript 无提示关闭窗口脚本
2009/08/17 Javascript
js DataSet数据源处理代码
2010/03/29 Javascript
Javascript中Array.prototype.map()详解
2014/10/22 Javascript
node.js中的require使用详解
2014/12/15 Javascript
JS根据生日算年龄的方法
2015/05/05 Javascript
JS实现的多张图片轮流播放幻灯片效果
2016/07/22 Javascript
jQuery实现自动输入email、时间和域名的方法
2016/08/24 Javascript
微信公众号支付H5调用支付解析
2016/11/04 Javascript
纯JS焦点图特效实例(可一个页面多用)
2016/12/07 Javascript
ionic中列表项增加和删除的实现方法
2017/01/22 Javascript
JavaScript Base64 作为文件上传的实例代码解析
2017/02/14 Javascript
HTML5+Canvas调用手机拍照功能实现图片上传(上)
2017/04/21 Javascript
在Django框架中运行Python应用全攻略
2015/07/17 Python
python数据结构之链表的实例讲解
2017/07/25 Python
利用python提取wav文件的mfcc方法
2019/01/09 Python
flask 框架操作MySQL数据库简单示例
2020/02/02 Python
Python脚本导出为exe程序的方法
2020/03/25 Python
python实现移动木板小游戏
2020/10/09 Python
Python之字符串的遍历的4种方式
2020/12/08 Python
css3 给页面加个半圆形导航条主要利用旋转和倾斜样式
2014/02/10 HTML / CSS
HTML5新增加的功能详解
2016/09/05 HTML / CSS
美国知名生活购物网站:Goop
2017/11/03 全球购物
设计部经理的岗位职责
2013/11/16 职场文书
愚人节活动策划方案
2014/03/11 职场文书
绿色城市实施方案
2014/03/19 职场文书
农民工工资承诺书范文
2014/03/31 职场文书
情况说明书格式范文
2014/05/06 职场文书
毕业生求职信
2014/06/10 职场文书
社保转移委托书范本
2014/10/08 职场文书
年度考核登记表个人总结
2015/03/06 职场文书
vue-cropper插件实现图片截取上传组件封装
2021/05/27 Vue.js