some doubts about the code for guassian kernel

[smallflyingpig]

when calculate the guassian kernel, I found this line in loss.py:
L2_distance = ((total0-total1)**2).sum(2)
why here is sum(2)?

[ChengYeung1222]

Sum up the tensor values in axis=2.