Flowchain - Method Chaining for TensorFlow
Project description
Flowchain - Method Chaining for TensorFlow
Extensive and simple tensor method chaining for TensorFlow
Only add 2 lines of code at the top of your codes!
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.2.tar.gz
(3.5 kB
view hashes)
Built Distribution
Close
Hashes for flowchain-0.0.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 28ed0e1310dd367fef551cfdf3735155cb2017276602a273aae01c21b79da896 |
|
MD5 | 5c13c9db2f564eae215032299124b33b |
|
BLAKE2b-256 | 9543115b3210e3b03f0893f43553eb516d3970d06230c4d4301c03c9e1e98ff1 |