[Urgent] Distribution Layers when data format is channels first
11 views
Skip to first unread message
Dushi Fdz
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
components_dist = component_layer(
tf.reshape(
params[..., num_components:],
tf.concat([tf.shape(params)[:-1], [num_components, -1]],
axis=0)))
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