本文重点
在数学中经常有加减乘除运算,在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