Skip to content

Unable to use heteroscedastic layers in tf.keras  #538

@keertika-11

Description

@keertika-11

TypeError: Dimension value must be integer or None or have an __index__ method, got value '<attribute 'shape' of 'numpy.generic' objects>' with type '<class 'getset_descriptor'>'

hidden_layer = tf.keras.layers.Dense(16)
inputs = tf.keras.Input((224,224,3))
out = hidden_layer(inputs)
output_layer = ed.layers.MCSoftmaxDense(num_classes=2)
output_layer(out)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions