tensorflow: 类numpy的api映射表

建立全零张量

tf.zeros(shape=(H, W), dtype=tf.float32)

np.zeros(shape=(H, W), dtype=np.float32)

建立全一张量

tf.ones(shape=(H, W), dtype=tf.float32)

np.ones(shape=(H, W), dtype=np.float32)

元素乘法

tf.mul(A, B)

A * B

矩阵乘法

tf.matmul(A, B)

np.dot(A * B)

张量形状

tf.shape(A)、A.get_shape()

A.shape()

塑形

tf.reshape(A, (H, W))

np.reshape(A, (H, W))

浮点化

tf.to_float(A)

float(A)

整型化

tf.to_int32(A)、tf.to_int64(A)

int(A)

张量的元素数量

tf.size(A)

np.size(A)

张量的阶

tf.rank(A)

np.ndim(A)

拼接

tf.concat(A, B)

np.concatenate(A, B)

求和

tf.reduce_sum(A, axis=0)

np.ndarray.sum(A, axis=0)

互相转换

tf.convert_to_tensor(A)