[Urgent] Distribution Layers when data format is channels first

10 views
Skip to first unread message

Dushi Fdz

unread,
Sep 4, 2021, 7:28:00 PM9/4/21
to TensorFlow Probability
Can someone please tell me how to change the source code in class MixtureSameFamily(DistributionLambda) in distribution layer when the data format is channels_first? https://github.com/tensorflow/probability/blob/9e50fa39f37216dd084c24aa3dd14ab0d4dee926/tensorflow_probability/python/layers/distribution_layer.py#L1396-L1520

The errors seem to be occurring in line 1483:


components_dist = component_layer(

tf.reshape(

params[..., num_components:],

tf.concat([tf.shape(params)[:-1], [num_components, -1]],

axis=0)))

Reply all
Reply to author
Forward
0 new messages