pytorch的张量相乘学习笔记有关mul与matmul函数

在学习过程中torch.mul是矩阵与矩阵里的元素相乘,然而torch.matmul与torch.mul不同的是matmul函数则是矩阵与矩阵的相乘

import  torch

data_1= [[1,2],[2,4]]
data_2= [[2,3],[3,4]]
tensor_1 = torch.tensor(data_1) # 转换为张量
tensor_2 = torch.tensor(data_2)

tensor_3 = torch.mul(tensor_1,tensor_2 )  # 函数里面的两个矩阵里对应的元素相乘
tensor_4 = torch.matmul(tensor_1,tensor_2) #函数里面的两个矩阵相乘

print(tensor_1)
print(tensor_2)
print(tensor_3)
print(tensor_4)

#tensor_1([[1, 2],
        [2, 4]])
tensor_2([[2, 3],
        [3, 4]])
tensor_3([[ 2,  6],
        [ 6, 16]])
tensor_4([[ 8, 11],
        [16, 22]])

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值