matlab axis square函数_1.4 TensorFlow2.1常用函数

1.4 TF常用函数

tf.cast(tensor,dtype=datatype)可以进行强制类型转换。

tf.reduce_min(tensor)和tf.reduce_max(tensor)将计算出张量中所有元素的最大值和最小值。

import tensorflow as tfx1 = tf.constant([1., 2., 3.], dtype=tf.float64)print("x1:", x1)x2 = tf.cast(x1, tf.int32)print("x2", x2)print("minimum of x2：", tf.reduce_min(x2))print("maxmum of x2:", tf.reduce_max(x2))

axis代表轴。以矩阵为例axis=0表示列而axis=1 表示行。tf.reduce_mean(tensor, axis=operating axis)被用来计算在axis方向上的随机数,但如果采用默认的axis,它将计算所有元素的均值。tf.reduce_sum(tensor, axis)计算合也是同理

import tensorflow as tfx = tf.constant([[1, 2, 3], [2, 2, 3]])print("x:", x)print("mean of x:", tf.reduce_mean(x))print("sum of x:", tf.reduce_sum(x, axis=1))

tf.Variable()可以将张量标记为可训练的。在回传环节被标记的张量可以记录梯度的信息。

import tensorflow as tfa = tf.ones([1, 3])b = tf.fill([1, 3], 3.)print("a:", a)print("b:", b)print("a+b:",