pytorch單維篩選 相乘的案例

m需要和篩選的結果維度相同

>0.5運行的結果與原來維度相同,結果是 0 1,0代不符合,1代表符合。

import torch
m=torch.Tensor([0.1,0.2,0.3]).cuda()
iou=torch.Tensor([0.5,0.6,0.7])
x= m * ((iou > 0.5).type(torch.cuda.FloatTensor))
print(x)

下面是把第一條與第二條變成瞭2:

import torch
m=torch.Tensor([0.1,0.2,0.3]).cuda()
iou=torch.Tensor([0.5,0.6,0.7])
a=iou > 0.5
m[[0,1,1]]=2
print(m)

下面的結果:0.2與0.3改為瞭2

import torch
m=torch.Tensor([0.1,0.2,0.3]).cuda()
iou=torch.Tensor([0.5,0.6,0.7])
a=iou > 0.5
m[a]=2
print(m)

補充:torch.Tensor的4種乘法

torch.Tensor有4種常見的乘法:*, torch.mul, torch.mm, torch.matmul. 本文拋磚引玉,簡單敘述一下這4種乘法的區別,具體使用還是要參照官方文檔。

點乘

a與b做*乘法,原則是如果a與b的size不同,則以某種方式將a或b進行復制,使得復制後的a和b的size相同,然後再將a和b做element-wise的乘法。

下面以*標量和*一維向量為例展示上述過程。

* 標量

Tensor與標量k做*乘法的結果是Tensor的每個元素乘以k(相當於把k復制成與lhs大小相同,元素全為k的Tensor).

>>> a = torch.ones(3,4)
>>> a
tensor([[1., 1., 1., 1.],
        [1., 1., 1., 1.],
        [1., 1., 1., 1.]])
>>> a * 2
tensor([[2., 2., 2., 2.],
        [2., 2., 2., 2.],
        [2., 2., 2., 2.]])

* 一維向量

Tensor與行向量做*乘法的結果是每列乘以行向量對應列的值(相當於把行向量的行復制,成為與lhs維度相同的Tensor). 註意此時要求Tensor的列數與行向量的列數相等。

>>> a = torch.ones(3,4)
>>> a
tensor([[1., 1., 1., 1.],
        [1., 1., 1., 1.],
        [1., 1., 1., 1.]])
>>> b = torch.Tensor([1,2,3,4])
>>> b
tensor([1., 2., 3., 4.])
>>> a * b
tensor([[1., 2., 3., 4.],
        [1., 2., 3., 4.],
        [1., 2., 3., 4.]])

Tensor與列向量做*乘法的結果是每行乘以列向量對應行的值(相當於把列向量的列復制,成為與lhs維度相同的Tensor). 註意此時要求Tensor的行數與列向量的行數相等。

>>> a = torch.ones(3,4)
>>> a
tensor([[1., 1., 1., 1.],
        [1., 1., 1., 1.],
        [1., 1., 1., 1.]])
>>> b = torch.Tensor([1,2,3]).reshape((3,1))
>>> b
tensor([[1.],
        [2.],
        [3.]])
>>> a * b
tensor([[1., 1., 1., 1.],
        [2., 2., 2., 2.],
        [3., 3., 3., 3.]])

* 矩陣

經Arsmart在評論區提醒,增補一個矩陣 * 矩陣的例子,感謝Arsmart的熱心評論!

如果兩個二維矩陣A與B做點積A * B,則要求A與B的維度完全相同,即A的行數=B的行數,A的列數=B的列數

>>> a = torch.tensor([[1, 2], [2, 3]])
>>> a * a
tensor([[1, 4],
        [4, 9]])

broadcast

點積是broadcast的。broadcast是torch的一個概念,簡單理解就是在一定的規則下允許高維Tensor和低維Tensor之間的運算。broadcast的概念稍顯復雜,在此不做展開,可以參考官方文檔關於broadcast的介紹. 在torch.matmul裡會有關於broadcast的應用的一個簡單的例子。

這裡舉一個點積broadcast的例子。在例子中,a是二維Tensor,b是三維Tensor,但是a的維度與b的後兩位相同,那麼a和b仍然可以做點積,點積結果是一個和b維度一樣的三維Tensor,運算規則是:若c = a * b, 則c[i,*,*] = a * b[i, *, *],即沿著b的第0維做二維Tensor點積,或者可以理解為運算前將a沿著b的第0維也進行瞭expand操作,即a = a.expand(b.size()); a * b。

>>> a = torch.tensor([[1, 2], [2, 3]])
>>> b = torch.tensor([[[1,2],[2,3]],[[-1,-2],[-2,-3]]])
>>> a * b
tensor([[[ 1,  4],
         [ 4,  9]],
        [[-1, -4],
         [-4, -9]]])
>>> b * a
tensor([[[ 1,  4],
         [ 4,  9]],
        [[-1, -4],
         [-4, -9]]])

其實,上面提到的二維Tensor點積標量、二維Tensor點積行向量,都是發生在高維向量和低維向量之間的,也可以看作是broadcast.

torch.mul

官方文檔關於torch.mul的介紹. 用法與*乘法相同,也是element-wise的乘法,也是支持broadcast的。

下面是幾個torch.mul的例子.

乘標量

>>> a = torch.ones(3,4)
>>> a
tensor([[1., 1., 1., 1.],
        [1., 1., 1., 1.],
        [1., 1., 1., 1.]])
>>> a * 2
tensor([[2., 2., 2., 2.],
        [2., 2., 2., 2.],
        [2., 2., 2., 2.]])

乘行向量

>>> a = torch.ones(3,4)
>>> a
tensor([[1., 1., 1., 1.],
        [1., 1., 1., 1.],
        [1., 1., 1., 1.]])
>>> b = torch.Tensor([1,2,3,4])
>>> b
tensor([1., 2., 3., 4.])
>>> torch.mul(a, b)
tensor([[1., 2., 3., 4.],
        [1., 2., 3., 4.],
        [1., 2., 3., 4.]])

乘列向量

>>> a = torch.ones(3,4)
>>> a
tensor([[1., 1., 1., 1.],
        [1., 1., 1., 1.],
        [1., 1., 1., 1.]])
>>> b = torch.Tensor([1,2,3]).reshape((3,1))
>>> b
tensor([[1.],
        [2.],
        [3.]])
>>> torch.mul(a, b)
tensor([[1., 1., 1., 1.],
        [2., 2., 2., 2.],
        [3., 3., 3., 3.]])

乘矩陣

例1:二維矩陣 mul 二維矩陣

>>> a = torch.tensor([[1, 2], [2, 3]])
>>> torch.mul(a,a)
tensor([[1, 4],
        [4, 9]])

例2:二維矩陣 mul 三維矩陣(broadcast)

>>> a = torch.tensor([[1, 2], [2, 3]])
>>> b = torch.tensor([[[1,2],[2,3]],[[-1,-2],[-2,-3]]])
>>> torch.mul(a,b)
tensor([[[ 1,  4],
         [ 4,  9]],
        [[-1, -4],
         [-4, -9]]])

torch.mm

官方文檔關於torch.mm的介紹. 數學裡的矩陣乘法,要求兩個Tensor的維度滿足矩陣乘法的要求.

例子:

>>> a = torch.ones(3,4)
>>> b = torch.ones(4,2)
>>> torch.mm(a, b)
tensor([[4., 4.],
        [4., 4.],
        [4., 4.]])

torch.matmul

官方文檔關於torch.matmul的介紹. torch.mm的broadcast版本.

例子:

>>> a = torch.ones(3,4)
>>> b = torch.ones(5,4,2)
>>> torch.matmul(a, b)
tensor([[[4., 4.],
         [4., 4.],
         [4., 4.]],
        [[4., 4.],
         [4., 4.],
         [4., 4.]],
        [[4., 4.],
         [4., 4.],
         [4., 4.]],
        [[4., 4.],
         [4., 4.],
         [4., 4.]],
        [[4., 4.],
         [4., 4.],
         [4., 4.]]])

同樣的a和b,使用torch.mm相乘會報錯

>>> torch.mm(a, b)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
RuntimeError: matrices expected, got 2D, 3D tensors at /pytorch/aten/src/TH/generic/THTensorMath.cpp:2065

以上這篇pytorch單維篩選 相乘的案例就是小編分享給大傢的全部內容瞭,希望能給大傢一個參考,也希望大傢多多支持WalkonNet。