Flowchain - Method Chaining for TensorFlow
Project description
Flowchain - Method Chaining for TensorFlow
Extensive and simple tensor method chaining for TensorFlow
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.4.tar.gz
(3.5 kB
view hashes)
Built Distribution
Close
Hashes for flowchain-0.0.4-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | f5da395c757e365b8c7f8726dc53680923f43a9bf6c88c757152487fc9c4e9a6 |
|
MD5 | 565088d5b5bef9e2f46488f7ed553122 |
|
BLAKE2b-256 | e572147e916867678608c67da64aba4ca33c037c23d1a2d3f06bb7854bdb81b1 |