代码拉取完成,页面将自动刷新
8191
redesigned bijector broadcast shape and bijector's dtype logic
已合并
What type of PR is this?
/kind feature
What does this PR do / why do we need it:
The PR redesigned the bijector class and made two major changes to the class:
- added a batch_shape to the bijector class. Now, transformed_distribution's broadcast_shape is calculated as the broadcast between the bijetor and the distribution.
- redesigned bijector's dtype logic, added restrictions to the dtype of the input values and the parameter(s) of the bijector.
Which issue(s) this PR fixes:
Fixes #
Special notes for your reviewers: