tf.py_func，自定义张量流函数仅应用于张量中的第一个元素

I am new to tensorflow and was playing around with a deep learning network. I wanted to do a custom rounding off on all the weights after each iteration. As the round function in tensorflow library doesn't give you the option to round the values down to a certain number of decimal points. So I wrote thisI am new to tensorflow and was playing around w