Пример нестандартной практической задачи, можете предлагать решения в комментариях.

Есть бинарная 2D маска. Нужно из неё сделать 2D float тензор, в котором есть хитрое выделение границы между значениями 0 и 1. Непосредственно на границе значения должны быть равны 1, а далее по формуле гауссового распределения падать до ~0 при удалении от границы на 50 пикселей.

Сможете сделать без двойного цикла? numpy, pytorch, opencv