@chenyuntc Thx… but I didn’t get it still. .unsqueeze(1) of the (64, 256) shaped B would result (64,1,256) tensor and then .repeat(1,10,1) will yield (64,10,256) which is the same shape with C (and thus becoming compatible with * operation btw B)
I thought this line executes for several times w/o any complaints but throws device-side error at some point.
If the operations with A, B, and C was the problem, Could you tell me more about it? Pls let me know…