facebookincubator / flowtorch

This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a community of users and contributors by focusing initially on complete infra and documentation for how to use and create components.
https://flowtorch.ai
MIT License
300 stars 21 forks source link

Fix domain and codomain for Compose #63

Closed stefanwebb closed 2 years ago

stefanwebb commented 2 years ago

Motivation

See #20

Changes proposed

Set the domain of Compose to the domain of the first Bijector, and codomain to the codomain of the last Bijector.