pytorch——pytorch常见运算详解

最新推荐文章于 2024-07-22 16:26:14 发布

chris_1996

最新推荐文章于 2024-07-22 16:26:14 发布

阅读量5.3k

点赞数

分类专栏： pytorch

本文链接：https://blog.csdn.net/weixin_45833008/article/details/108699280

版权

本文详细介绍了PyTorch中的常见运算，包括矩阵与标量的运算、哈达玛积、矩阵乘法、幂与开方、对数运算、近似值运算以及剪裁运算。特别强调了矩阵乘法的不同实现方式和多维张量的矩阵运算规则，还提到了在对抗生成网络中剪裁运算的应用。

摘要由CSDN通过智能技术生成

pytorch常见运算详解

这个是矩阵（张量）每一个元素与标量进行操作。

import torch
a = torch.tensor([1,2])
print(a+1)
>>> tensor([2, 3])

这个就是两个相同尺寸的张量相乘，然后对应元素的相乘就是这个哈达玛积，也成为element wise。

a = torch.tensor([1,2])
b = torch.tensor([2,3])
print(a*b)
print(torch.mul(a,b))
>>> tensor([2, 6])
>>> tensor([2, 6])

这个torch.mul()和*是等价的。
当然，除法也是类似的：

a = torch.tensor([1.,2.])
b = torch.tensor([2.,3.])
print(a/b)
print(<

关注