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) |
tensorflow: 类numpy的api映射表
最新推荐文章于 2024-07-10 10:08:47 发布