google-deepmind / chex

https://chex.readthedocs.io
Apache License 2.0
769 stars 44 forks source link

Add Dimensions.size to get flat size from named shape. #339

Closed copybara-service[bot] closed 6 months ago

copybara-service[bot] commented 6 months ago

Add Dimensions.size to get flat size from named shape.