lucid
lucid copied to clipboard
collapse_alpha needs known shape
Let's assert the tensor shape coming into collapse alpha to be known at graph creation time.
Currently you can get an exception if you ask for a collapse_alpha transform to be applied after a transform that stochastically changes t_image's shape (such as random_scale).