luodahei:

[128,64] and [128,64,1,1]


Is a bit different. It’s the same amount of numbers within the tensor but the two last dimensions are “empty”. You can reshape the first one to be the same size as the second one. Perhaps this view function 1 could help you.
If it still is tricky, post some code / github link and we can take it from there