pytorch矩阵乘法
torch.mm(mat1, mat2) performs a matrix multiplication of mat1 and mat2
a = torch.randint(0, 5, (2, 3)) # tensor([[3, 3, 2], # [2, 2, 2]]) b = torch.randint(0, 6, (3, 1)) # tensor([[1], # [4], # [5]]) torch.mm(a, b) # tensor([[11], # [17]])
torch.mul(input, other) multiplies each element of the 'input' with the scalar in 'other' and returns a new resulting tensor.
a = torch.tensor([[1, 2, 3], [4, 5, 6]]) b = torch.tensor([1, 2, 3]) c = torch.tensor([[1], [2], [3]]) torch.mul(a, 10) # tensor([[10, 20, 30], # [40, 50, 60]]) torch.mul(a, b) # tensor([[ 1, 4, 9], # [ 4, 10, 18]]) torch.mul(b, c) # tensor([[1, 2, 3], # [2, 4, 6], # [3, 6, 9]])