dw做音乐网站黑马培训

本文重点

在数学中经常有加减乘除运算，在tensor中也不例外，也有类似的运算，本节课程我们将学习tensor中的运算

常见运算

加法+或者add

import torch
import numpy as np
a=torch.rand(16,3,28,28)
b=torch.rand(1,3,28,28)
print(a+b)
import torch
import numpy as np
a=torch.rand(16,3,28,28)
b=torch.rand(1,3,28,28)
print(torch.add(a,b))

注意，这里面维度并不是一样的，此时会使用前面学习的pytorch中的广播机制

减法sub或者-

import torch
import numpy as np
a=torch.rand(16,3,28,28)
b=torch.rand(1,3,28,28)
print(a-b)
import torch
import numpy as np
a=torch.rand(16,3,28,28)
b=torch.rand(1,3,28,28)
print(torch.sub(a,b))

相乘mul或者*（对位相乘）

import torch
import numpy as np
a=torch.rand(16,3,28,28)
b=torch.rand(1,3,28,28)
print(a*b)
import torch
import