Я знаю, что есть функция pad, которая создает вид на тензор, дополненный нулями. Вместо этого я хотел бы иметь (примеры в 2D):
Введите:
1 2 3
4 5 6
7 8 9
Выход дополнен одним элементом со всех сторон:
1 1 2 3 3
1 1 2 3 3
4 4 5 6 6
7 7 8 9 9
7 7 8 9 9
Есть ли простой или прямой способ сделать это?
Лучшее, что я могу придумать, - это вручную установить для всех угловых областей правильное значение, что, вероятно, является довольно неэффективным и довольно сложным для написания кода.