如何在PyTorch中乘以两个张量轴?

2024-04-27 22:45:53 发布

您现在位置:Python中文网/ 问答频道 /正文

我有两个形状的三维张量:

a = torch.full([1495, 110247, 1], 0.5)
b = torch.full([1495, 110247, 2], 1)

我想将它们相乘,使前两个维度保持不变,第三个维度是第三个维度ab的标量积


Tags: torchfull形状标量