数学运算:

1
2
3
4
5
6
7
8
9
a = torch.tensor([[1, 2, 3], [4, 5, 6]])
 
b = torch.tensor([[7, 8, 9], [10, 11, 12]])
 
tensor([[1, 2, 3],
        [4, 5, 6]])
 
tensor([[ 7,  8,  9],
        [10, 11, 12]])
  • torch.add(tensor1, tensor2): 张量相加。
1
2
3
result = torch.add(a, b)
tensor([[ 8, 10, 12],
        [14, 16, 18]])

  

  • torch.sub(tensor1, tensor2): 张量相减。
  • torch.mul(tensor1, tensor2): 张量相乘。
1
2
3
torch.mul(a,b)
tensor([[ 7, 16, 27],
        [40, 55, 72]])

  

  • torch.div(tensor1, tensor2): 张量相除
posted on   黑逍逍  阅读(34)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 全程不用写代码,我用AI程序员写了一个飞机大战
· DeepSeek 开源周回顾「GitHub 热点速览」
· 记一次.NET内存居高不下排查解决与启示
· 物流快递公司核心技术能力-地址解析分单基础技术分享
· .NET 10首个预览版发布:重大改进与新特性概览!



点击右上角即可分享
微信分享提示