Hi,
Can the eight 0.4 in the first column of tensor1 be multiplied by the eight rows of tensor2? As shown by the red and blue circles. Require 0.4 to be multiplied by the three numbers in the first row, and so on. How should this be achieved?
tensor1([[0.5000, 0.5000],
        [0.5000, 0.5000],
        [0.5000, 0.5000],
        [0.5000, 0.5000],
        [0.5000, 0.5000],
        [0.5000, 0.5000],
        [0.5000, 0.5000],
        [0.5000, 0.5000]], device='cuda:0')
tensor2([[0.0794, 0.0728, 0.8477],
        [0.1744, 0.0727, 0.7529],
        [0.6121, 0.2235, 0.1644],
        [0.0610, 0.0458, 0.8932],
        [0.5754, 0.1586, 0.2660],
        [0.8185, 0.1012, 0.0802],
        [0.4628, 0.1336, 0.4036],
        [0.3557, 0.0482, 0.5960]], device='cuda:0')


微信图片_20210715154451786×556 59.7 KB