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]]])