some doubts about the code for guassian kernel
smallflyingpig opened this issue · 1 comments
smallflyingpig commented
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 commented
Sum up the tensor values in axis=2.