Flowchain - Method Chaining for TensorFlow
Project description
Flowchain - Method Chaining for TensorFlow
Extensive and simple tensor method chaining for TensorFlow
This package requires Python >= 3.7 and uses some Python magics for extensive injection. Therefore, there could be some unexpected behaviour among method chaining. If you have experienced such behaviours, please feel free to open new issue in this repo!
Currently, I am still finding a way to support code completion for IDEs like PyCharm, Visual Studio Code, etc.
Install via:
pip install flowchain
Add only 2 lines of code at the top of your code!
from flowchain import enable_tensor_chaining
enable_tensor_chaining() # this does everything for you.
This package makes following approach possible:
# before
x = tf.abs(lhs - rhs)
x = tf.reduce_sum(x, 1)
x = tf.argmin(x, output_type=tf.int32)
# after
x = (lhs - rhs).abs().reduce_sum(1).argmin(output_type=tf.int32)
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
flowchain-0.0.6.tar.gz
(3.7 kB
view hashes)
Built Distribution
Close
Hashes for flowchain-0.0.6-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | f6c1653703706becbf1a3bfda3f8f97b558ee8398d5ed523bfcf8ce2c69a0d25 |
|
MD5 | 3f57125d72776b74b5fed10c0c5e0c95 |
|
BLAKE2b-256 | ed079892c2e866710c7ae8806196a1427b596e37e69575b299c71121ed267ab9 |