PyTorch ---- torch.nn.function.pad 函數(shù)用法(補(bǔ)充維度上的數(shù)值)

1.二維數(shù)組:對(duì)最內(nèi)部元素左側(cè)增加元素(例如 1 的左側(cè))

a = torch.tensor([[1, 2, 3, 4], [1, 2, 3, 4]])
a1 = torch.nn.functional.pad(a, pad=(1, 0, 0, 0), mode='constant', value=1)
print("a = ", a)
print("a1 = ", a1)

運(yùn)行結(jié)果:

a =  tensor([[1, 2, 3, 4],
        [1, 2, 3, 4]])
a1 =  tensor([[1, 1, 2, 3, 4],
        [1, 1, 2, 3, 4]])

2.二維數(shù)組:對(duì)最內(nèi)部元素右側(cè)增加元素(例如 4 右側(cè))

a = torch.tensor([[1, 2, 3, 4], [1, 2, 3, 4]])
a1 = torch.nn.functional.pad(a, pad=(0, 1, 0, 0), mode='constant', value=1)
print("a = ", a)
print("a1 = ", a1)

運(yùn)行結(jié)果:

a =  tensor([[1, 2, 3, 4],
        [1, 2, 3, 4]])
a1 =  tensor([[1, 2, 3, 4, 1],
        [1, 2, 3, 4, 1]])

3.二維數(shù)組:對(duì)最內(nèi)部一維數(shù)組左側(cè)增加元素(例如 [1, 2, 3, 4] 左側(cè))

a = torch.tensor([[1, 2, 3, 4], [1, 2, 3, 4]])
a1 = torch.nn.functional.pad(a, pad=(0, 0, 1, 0), mode='constant', value=1)
print("a = ", a)
print("a1 = ", a1)

運(yùn)行結(jié)果:

a =  tensor([[1, 2, 3, 4],
        [1, 2, 3, 4]])
a1 =  tensor([[1, 1, 1, 1],
        [1, 2, 3, 4],
        [1, 2, 3, 4]])

4.二維數(shù)組:對(duì)最內(nèi)部一維數(shù)組右側(cè)增加元素(例如 [1, 2, 3, 4] 右側(cè))

a = torch.tensor([[1, 2, 3, 4], [1, 2, 3, 4]])
a1 = torch.nn.functional.pad(a, pad=(0, 0, 0, 1), mode='constant', value=1)
print("a = ", a)
print("a1 = ", a1)

運(yùn)行結(jié)果:

a =  tensor([[1, 2, 3, 4],
        [1, 2, 3, 4]])
a1 =  tensor([[1, 2, 3, 4],
        [1, 2, 3, 4],
        [1, 1, 1, 1]])

5.三維數(shù)組:對(duì)最內(nèi)部元素左側(cè)增加元素(例如 1 左側(cè))

a = torch.tensor([[[1, 2, 3, 4], [5, 6, 7, 8]], [[1, 2, 3, 4], [5, 6, 7, 8]]])
a1 = torch.nn.functional.pad(a, pad=(1, 0, 0, 0, 0, 0), mode='constant', value=1)
print("a = ", a)
print("a1 = ", a1)

運(yùn)行結(jié)果:

a =  tensor([[[1, 2, 3, 4],
         [5, 6, 7, 8]],


        [[1, 2, 3, 4],
         [5, 6, 7, 8]]])
a1 =  tensor([[[1, 1, 2, 3, 4],
         [1, 5, 6, 7, 8]],


        [[1, 1, 2, 3, 4],
         [1, 5, 6, 7, 8]]])

6.三維數(shù)組:對(duì)最內(nèi)部元素右側(cè)增加元素(例如 4 右側(cè))

a = torch.tensor([[[1, 2, 3, 4], [5, 6, 7, 8]], [[1, 2, 3, 4], [5, 6, 7, 8]]])
a1 = torch.nn.functional.pad(a, pad=(0, 1, 0, 0, 0, 0), mode='constant', value=1)
print("a = ", a)
print("a1 = ", a1)

運(yùn)行結(jié)果:

a =  tensor([[[1, 2, 3, 4],
         [5, 6, 7, 8]],


        [[1, 2, 3, 4],
         [5, 6, 7, 8]]])
a1 =  tensor([[[1, 2, 3, 4, 1],
         [5, 6, 7, 8, 1]],


        [[1, 2, 3, 4, 1],
         [5, 6, 7, 8, 1]]])

7.三維數(shù)組:對(duì)最內(nèi)部一維數(shù)組左側(cè)增加元素(例如 [1, 2, 3, 4] 左側(cè))

a = torch.tensor([[[1, 2, 3, 4], [5, 6, 7, 8]], [[1, 2, 3, 4], [5, 6, 7, 8]]])
a1 = torch.nn.functional.pad(a, pad=(0, 0, 1, 0, 0, 0), mode='constant', value=1)
print("a = ", a)
print("a1 = ", a1)

運(yùn)行結(jié)果:

a =  tensor([[[1, 2, 3, 4],
         [5, 6, 7, 8]],


        [[1, 2, 3, 4],
         [5, 6, 7, 8]]])
a1 =  tensor([[[1, 1, 1, 1],
         [1, 2, 3, 4],
         [5, 6, 7, 8]],


        [[1, 1, 1, 1],
         [1, 2, 3, 4],
         [5, 6, 7, 8]]])

8.三維數(shù)組:對(duì)最內(nèi)部一維數(shù)組右側(cè)增加元素(例如 [5, 6, 7, 8] 右側(cè))

a = torch.tensor([[[1, 2, 3, 4], [5, 6, 7, 8]], [[1, 2, 3, 4], [5, 6, 7, 8]]])
a1 = torch.nn.functional.pad(a, pad=(0, 0, 0, 1, 0, 0), mode='constant', value=1)
print("a = ", a)
print("a1 = ", a1)

運(yùn)行結(jié)果:

a =  tensor([[[1, 2, 3, 4],
         [5, 6, 7, 8]],
        [[1, 2, 3, 4],
         [5, 6, 7, 8]]])
a1 =  tensor([[[1, 2, 3, 4],
         [5, 6, 7, 8],
         [1, 1, 1, 1]],
        [[1, 2, 3, 4],
         [5, 6, 7, 8],
         [1, 1, 1, 1]]])

9.三維數(shù)組:對(duì)最內(nèi)部二維數(shù)組左側(cè)增加元素(例如 [[1, 2, 3, 4], [5, 6, 7, 8]] 左側(cè))

a = torch.tensor([[[1, 2, 3, 4], [5, 6, 7, 8]], [[1, 2, 3, 4], [5, 6, 7, 8]]])
a1 = torch.nn.functional.pad(a, pad=(0, 0, 0, 0, 1, 0), mode='constant', value=1)
print("a = ", a)
print("a1 = ", a1)
運(yùn)行結(jié)果:

a =  tensor([[[1, 2, 3, 4],
         [5, 6, 7, 8]],


        [[1, 2, 3, 4],
         [5, 6, 7, 8]]])
a1 =  tensor([[[1, 1, 1, 1],
         [1, 1, 1, 1]],


        [[1, 2, 3, 4],
         [5, 6, 7, 8]],


        [[1, 2, 3, 4],
         [5, 6, 7, 8]]])

10.三維數(shù)組:對(duì)最內(nèi)部二維數(shù)組左側(cè)增加元素 x2(例如 [[1, 2, 3, 4], [5, 6, 7, 8]] 左側(cè))

a = torch.tensor([[[1, 2, 3, 4], [5, 6, 7, 8]], [[1, 2, 3, 4], [5, 6, 7, 8]]])
a1 = torch.nn.functional.pad(a, pad=(0, 0, 0, 0, 2, 0), mode='constant', value=1)
print("a = ", a)
print("a1 = ", a1)

運(yùn)行結(jié)果:

a =  tensor([[[1, 2, 3, 4],
         [5, 6, 7, 8]],


        [[1, 2, 3, 4],
         [5, 6, 7, 8]]])
a1 =  tensor([[[1, 1, 1, 1],
         [1, 1, 1, 1]],


        [[1, 1, 1, 1],
         [1, 1, 1, 1]],


        [[1, 2, 3, 4],
         [5, 6, 7, 8]],


        [[1, 2, 3, 4],
         [5, 6, 7, 8]]])

11.三維數(shù)組:對(duì)最內(nèi)部二維數(shù)組右側(cè)增加元素 (例如 [[11, 22, 33, 44], [55, 66, 77, 88]] 右側(cè))

a = torch.tensor([[[1, 2, 3, 4], [5, 6, 7, 8]], [[11, 22, 33, 44], [55, 66, 77, 88]]])
a1 = torch.nn.functional.pad(a, pad=(0, 0, 0, 0, 0, 1), mode='constant', value=1)
print("a = ", a)
print("a1 = ", a1)

運(yùn)行結(jié)果:

a =  tensor([[[ 1,  2,  3,  4],
         [ 5,  6,  7,  8]],


        [[11, 22, 33, 44],
         [55, 66, 77, 88]]])
a1 =  tensor([[[ 1,  2,  3,  4],
         [ 5,  6,  7,  8]],


        [[11, 22, 33, 44],
         [55, 66, 77, 88]],


        [[ 1,  1,  1,  1],
         [ 1,  1,  1,  1]]])

12.三維數(shù)組:對(duì)最內(nèi)部二維數(shù)組右側(cè)增加元素 x2 (例如 [[11, 22, 33, 44], [55, 66, 77, 88]] 右側(cè))

a = torch.tensor([[[1, 2, 3, 4], [5, 6, 7, 8]], [[11, 22, 33, 44], [55, 66, 77, 88]]])
a1 = torch.nn.functional.pad(a, pad=(0, 0, 0, 0, 0, 2), mode='constant', value=1)
print("a = ", a)
print("a1 = ", a1)

運(yùn)行結(jié)果:

a =  tensor([[[ 1,  2,  3,  4],
         [ 5,  6,  7,  8]],


        [[11, 22, 33, 44],
         [55, 66, 77, 88]]])
a1 =  tensor([[[ 1,  2,  3,  4],
         [ 5,  6,  7,  8]],


        [[11, 22, 33, 44],
         [55, 66, 77, 88]],


        [[ 1,  1,  1,  1],
         [ 1,  1,  1,  1]],


        [[ 1,  1,  1,  1],
         [ 1,  1,  1,  1]]])
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書(shū)系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

友情鏈接更多精彩內(nèi)容