Tensorflow包装器就像numpy(非常接近的API)一样?

时间:2017-06-01 18:48:07

标签: python numpy tensorflow

是否存在Tensorflow的库/包装器,它允许您使用相同的API /语法执行numpy中的所有操作?换句话说,人们几乎可以用np.替换每个tf.并获得非常接近的功能?

然后,可以做一些像这样的时髦的东西(我通过继承tf.Variable来做到这一点):

a = Tens([[1, -2, 3]], name='a')
b = Tens([[2], [4], [-1]], name='b')

dot_product = a @ b
vector_sum = a + b.T

sess = tf.InteractiveSession()
init_op = tf.global_variables_initializer()
sess.run(init_op)

print(dot_product.eval())
print(vector_sum.eval())

>>> [[-9]]
>>> [[3 2 2]]

能够做一个简单的替换numpy脚本并获得TF处理的所有乐趣将是很酷的。

如果这也对您有吸引力,请在评论中告诉我。

0 个答案:

没有答案