TensorFlow和numpy常用函数（二）

1.条件语句where

条件语句为真返回A，条件语句为假返回B：

tf.where(条件语句, 真返回A, 假返回B)

例如求两张量每个元素的最大值：

import tensorflow as tf

a = tf.constant([1, 2, 3, 1, 1])
b = tf.constant([0, 1, 3, 4, 5])
c = tf.where(tf.greater(a, b), a, b)  # 若a>b，返回a对应位置的元素，否则返回b对应位置的元素
print("c：", c)

输出结果为：

c： tf.Tensor([1 2 3 4 5], shape=(5,), dtype=int32)

2.随机数生成0~1之间np.random.RandomState.rand

返回一个[0, 1)之间的随机数：

np.random.RandomState.rand(维度)
注： 维度为空时返回标量

例如：

import numpy as np

rdm = np.random