A library for annotating and checking the shapes of tensors.
Reason this release was yanked:
Broken dependency.
Project description
check_shapes
check_shapes
is a library for annotating and checking tensor shapes.
For example:
import tensorflow as tf
from gpflow.experimental.check_shapes import check_shapes
@tf.function
@check_shapes(
"features: [batch..., n_features]",
"weights: [n_features]",
"return: [batch...]",
)
def linear_model(features: tf.Tensor, weights: tf.Tensor) -> tf.Tensor:
return tf.einsum("...i,i -> ...", features, weights)
For more information see our documentation.
Installation
The recommended way to install check_shapes
is from pypi:
pip install check_shapes
From source
To develop check_shapes
, check it out from GitHub:
git clone git@github.com:GPflow/check_shapes.git
We use Poetry to install and manage dependencies. Follow their instructions for how to install Poetry itself. Then:
cd check_shapes
poetry install
To check you installation run our tests:
poetry run task test
For testing with different versions of Python and dependencies, see the poetryenv
script.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
check_shapes-1.1.0.tar.gz
(32.6 kB
view hashes)
Built Distribution
Close
Hashes for check_shapes-1.1.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 2487c23a9b5925b0030e31d6c52f33ddbf653f4991c52d65cd9a475e1f7d9405 |
|
MD5 | ea14e8ddbd544a807894bd182af229d6 |
|
BLAKE2b-256 | 251533b482c7036fb9996fb82353a1a1d8f4ecf97f36387aaa63489358cb0bcc |