License

Hello everyone,

First of all, thanks for the material! 
I tried to look for the licensing terms under which the software is distributed, but I did not find anything. Since it is inspired by a Tensorflow function, I guess that everything is under Apache 2.0.
Am I correct? :-)

Thanks a lot,
Matteo