search

google-deepmind / rlax

https://rlax.readthedocs.io
Apache License 2.0
1.24k stars 85 forks source link

Fix a bug in tree_split_leaves(): squeeze the right axis in case of keepdim=False. #94

Closed copybara-service[bot] closed 2 years ago

copybara-service[bot] commented 2 years ago

Fix a bug in tree_split_leaves(): squeeze the right axis in case of keepdim=False.