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.5.tar.gz
(3.7 kB
view hashes)
Built Distribution
Close
Hashes for flowchain-0.0.5-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 9dba70996c96816b3e2abe62377376d5f156920a25127ef33a632e098120938d |
|
MD5 | 7b94f0ac3908cf81c56e782a40610f08 |
|
BLAKE2b-256 | 56276ede65b3942f5ff08500f1172ed1e2362c6d063541a22e98f04cdacc24ea |