tensorflow: 类numpy的api映射表

Annotations	tensorflow api	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)

---

---