tensorflow / lattice

Lattice methods in TensorFlow

tf.TensorShape should get only one argument

devavratTomar opened this issue 4 years ago · comments

Devavrat Tomar 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.