Tensorflow 常用函数总结

本文介绍TensorFlow中的基本数学运算操作,包括四则运算、矩阵运算及数理统计等,并概述了tf.nn库中用于实现卷积神经网络(CNN)、循环神经网络(RNN)和其他常用神经网络组件的功能函数。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

四则运算:

    平方:tf.square(),开方:tf.sqrt()

    tf.add()、tf.div()、tf.mod()、tf.abs() tf.multiply()  tf.subtract()  tf.negative()

矩阵运算:

    tf.matmul(a,b) //矩阵乘法

数理统计:

    均值:tf.reduce_mean(x,reduction_indices=[1]//维度),求和:tf.reduce_sum

    最大值/最小值: tf.reduce_max/min

    最大值下标 :tf.argmax

tf.cast(x,数据类型)//强制类型转换

tf.nn库:

    tf.nn,conv2d(x,w,strides,padding) //卷积

    tf.nn.max_pool(x,ksize,strides,padding) //池化

    tf.nn.relu()  tf.nn.lrn()     tf.nn.softmax()

    tf.nn.sparse_softmax_cross_entropy_with_logits(logits,labels) //softmax + cross_entropy 注意label不是one-hot的,然后接上一个tf.reduce_mean()

    tf.nn.in_top_k(logits,label,k) //判断label是不是在logits前k大的值里,返回值是True和False列表

初始化方法:

    tf.truncated_normal(shape,stddev) //阶段正态分布初始化权重,避免完全对称

tf.add_to_collection('loss',v1) //将变量添加到列表

tf.get_collection('loss') //从列表中取出全部‘loss’命名的变量列表

tf.add_n(tf.get_collection('loss')) //将全部loss取出后求和

tensor.get_shape().as_list()

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值