是否存在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处理的所有乐趣将是很酷的。
如果这也对您有吸引力,请在评论中告诉我。