TensorFlow中的常见函数

TensorFlow中的常见函数

import tensorflow as tf
1.tf.cast(张量名，dtype=数据类型)
效果为强制tensor转换为该数据类型，例如下代码：

x1 = tf.constant([1., 2., 3.], dtype=tf.float64)
print("x1:", x1)
x2 = tf.cast(x1, tf.int32)
print("x2", x2)

输出结果为：

x1: tf.Tensor([1. 2. 3.], shape=(3,), dtype=float64)
x2 tf.Tensor([1 2 3], shape=(3,), dtype=int32)

即将x1转化为int32型。

2、tf.reduce_min(张量名)，tf.reduce_max(张量名)
用来计算张量维度上的最小值和最大值。代码如下：

print("minimum of x2：", tf.reduce_min(x2))
print("maxmum of x2:", tf.reduce_max(x2))

输出结果为：

minimum of x2： tf.Tensor(1, shape=(), dtype=int32)
maxmum of x2: tf.Tensor(3, shape=(), dtype=int32)

3、axis=操作轴
axis=0表示在列上操作，axis=1表示在行上操作。
可见实例程序如下：

 x = tf.constant([[1, 2, 3], [2, 2, 3]])
print("x:", x)
print("mean of x:", tf.reduce_mean(x))  # 求x中所有数的均值(结果取整数位)
print("sum of x:", tf.reduce_sum(x, axis=1))  # 求每一行的和

输出为：

x: tf.Tensor(
[[1 2 3]
 [2 2 3]], shape=(2, 3), dtype=int32)
mean of x: tf.Tensor(2, shape=(), dtype=int32)
sum of x: tf.Tensor([6 7