tensorflow中一些常见的函数(持续更新)
程序员文章站
2022-04-20 08:32:06
...
1.tf.cast()函数的作用是执行 tensorflow 中张量数据类型转换,比如读入的图片如果是int8类型的,一般在要在训练前把图像的数据格式转换为float32。
tf.cast(x, dtype, name=None)
第一个参数 x: 待转换的数据(张量)
第二个参数 dtype: 目标数据类型
第三个参数 name: 可选参数,定义操作的名称
**2.**tf.reduce_mean 函数用于计算张量tensor沿着指定的数轴(tensor的某一维度)上的的平均值,主要用作降维或者计算tensor(图像)的平均值。
reduce_mean(input_tensor,
axis=None,
keep_dims=False,
name=None,
reduction_indices=None)
第一个参数input_tensor: 输入的待降维的tensor;
第二个参数axis: 指定的轴,如果不指定,则计算所有元素的均值,若axis=0,则计算垂直方向上的平均值,若axis=1,则计算水平方向上的平均值;
第三个参数keep_dims:是否降维度,设置为True,输出的结果保持输入tensor的形状,设置为False,输出结果会降低维度;
第四个参数name: 操作的名称;
第五个参数 reduction_indices:在以前版本中用来指定轴,已弃用;
3.
tf.reduce_sum将矩阵的元素相加,可以按列,也可以按行,或者先按列后按行
reduce_sum(
input_tensor,
axis=None,
keep_dims=False,
name=None,
reduction_indices=None
)
input_tensor:输入的张量数组
axis:求和的维度,指定的轴,如果不指定,则计算所有元素的和,若axis=0,则计算垂直方向上的和,若axis=1,则计算水平方向上的和
keepdims:是否保持维度不变