Skip to main content

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.8.tar.gz (4.0 kB view hashes)

Uploaded Source

Built Distribution

flowchain-0.0.8-py3-none-any.whl (4.7 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page