How can i use this function as a Lambda layer in my neural network

[arbaz52]

tensorflow_unpooling/unpool.py

Line 20 in e45e03d

pooled_ = tf.reshape(pooled, [input_shape[0] * input_shape[1] * input_shape[2] * input_shape[3]])

Shows me the following error when i create the model:
pooled_ = tf.reshape(pooled, [input_shape[0] * input_shape[1] * input_shape[2] * input_shape[3]])
TypeError: unsupported operand type(s) for *: 'NoneType' and 'int'

[han-yeol]

tensorflow_unpooling/unpool.py

Line 20 in e45e03d

pooled_ = tf.reshape(pooled, [input_shape[0] * input_shape[1] * input_shape[2] * input_shape[3]])

Shows me the following error when i create the model:
pooled_ = tf.reshape(pooled, [input_shape[0] * input_shape[1] * input_shape[2] * input_shape[3]])
TypeError: unsupported operand type(s) for *: 'NoneType' and 'int'

same error is occurred.
Did you solve that?

[arbaz52]

I stopped working on it a while ago, but I was trying to use it in my project to build an attention layer.
Please check the following repository to see if you can find out how I used it, that might help you fix the error you're facing.
https://github.com/arbaz52/attention-network
If you cannot find it there, I have worked on Tensorflow in past, feel free to check my repositories to find one that fits your situation.