search

tensorflow / lattice

Lattice methods in TensorFlow
Apache License 2.0
518 stars 94 forks source link

tf.TensorShape should get only one argument #46

Closed devavratTomar closed 4 years ago

devavratTomar commented 4 years ago

I received an error when https://github.com/tensorflow/lattice/blob/master/tensorflow_lattice/python/parallel_combination_layer.py#L152 this line was executed by me.

it should be replaced by tf.TensorShape([None, ...])

wbakst commented 4 years ago

Thanks for catching this!

mmilanifard commented 4 years ago

Fixed in 2.0.4.