Commit 8a3b8e1a authored by Eugene Brevdo's avatar Eugene Brevdo Committed by gunan
Browse files

Simplify bijector API: remove the need for batch_ndims. 

batch_ndims never needs to be provided explicitly.  the two bijectors that use it,
Affine and AffineLinearOp, can both derive it from their arguments.

This brings us back to a more TITO (Tensor-in, Tensor-out) form for the bijector
API.
Change: 146408773
parent 37132afd
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment