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