tensorflow 常见方法总结
迪丽瓦拉
2025-06-01 06:50:25
0

矩阵相乘:

tf.matmul(x,y) 矩阵相乘:A 的shape为(2,3),B的shape为(3,4),则结果shape为(2,4)

矩阵和矩阵元素的操作:

矩阵元素相加:

tf.math.add(x,y)等价于x + y

矩阵相减:

tf.math.subtract(x,y)等价于x-y

矩阵元素相乘:

tf.math.multiply(x,y) 等价于:x*y

矩阵元素相除:

tf.math.devide(x,y)等价于x/y

矩阵元素绝对值:

tf.math.abs(x)

比较操作:

tf.math.maximum(a,b)

tf.math.minimum(a,b)

tf.math.equal(a,b)

tf.math.equal(a,3)

tf.math.equal(3,3)

tf.math.equal(a,a)

tf.math.greater(a,3)

tf.math.greater_equal(a,3)

矩阵和数字的操作:

矩阵元素数乘:3*a 等价于a*3

矩阵元素数加:3+a 等价于a+3

矩阵元素数减:3-a等价于a-3

矩阵元素数除a/3,3/a

矩阵元素的类似计算操作都在tf.math包中

特别的:

最大值:

tf.math.reduce_max(a,axis=1,keepdims=True)将最后一维最大值,可以选择保留维度或压缩掉

最大值对应的位置indices:

tf.argmax(predict_result["logits"],axis=-1)将最后一维进行压缩求最大值对应的position

求top_k最大值:

tf.math.top_k(x)

tf.where操作:

将向量中大于0.5的设置为1,小于0.5的设置为0

tf.where(A>0.5,x=1,y=0)表示将A>0.5中条件为true的元素设置为x,为false的设置为y

相关内容

热门资讯

linux入门---制作进度条 了解缓冲区 我们首先来看看下面的操作: 我们首先创建了一个文件并在这个文件里面添加了...
C++ 机房预约系统(六):学... 8、 学生模块 8.1 学生子菜单、登录和注销 实现步骤: 在Student.cpp的...
A.机器学习入门算法(三):基... 机器学习算法(三):K近邻(k-nearest neigh...
数字温湿度传感器DHT11模块... 模块实例https://blog.csdn.net/qq_38393591/article/deta...
有限元三角形单元的等效节点力 文章目录前言一、重新复习一下有限元三角形单元的理论1、三角形单元的形函数(Nÿ...
Redis 所有支持的数据结构... Redis 是一种开源的基于键值对存储的 NoSQL 数据库,支持多种数据结构。以下是...
win下pytorch安装—c... 安装目录一、cuda安装1.1、cuda版本选择1.2、下载安装二、cudnn安装三、pytorch...
MySQL基础-多表查询 文章目录MySQL基础-多表查询一、案例及引入1、基础概念2、笛卡尔积的理解二、多表查询的分类1、等...
keil调试专题篇 调试的前提是需要连接调试器比如STLINK。 然后点击菜单或者快捷图标均可进入调试模式。 如果前面...
MATLAB | 全网最详细网... 一篇超超超长,超超超全面网络图绘制教程,本篇基本能讲清楚所有绘制要点&#...