Cómo rellenar los límites del tensor de entrada con cero en PyTorch

En este artículo, discutiremos cómo rellenar los límites del tensor de entrada con cero en Python usando PyTorch .

método torch.nn.ZeroPad2D()

Este método ha aceptado el tamaño del relleno como entrada y devuelve un nuevo tensor con límites. Los límites pueden ser iguales o diferentes en todos los lados (izquierda, derecha, arriba, abajo) . podemos aumentar la altura y el ancho de un tensor acolchado usando arriba+abajo e izquierda+derecha respectivamente. La siguiente sintaxis se usa para rellenar los límites del tensor de entrada con cero.

Sintaxis: torch.nn.ZeroPad2d(pad)

Parámetro:

• pad (int, tuple): este es el tamaño del relleno. El tamaño del relleno es un número entero o una tupla.

Retorno: este método devuelve un nuevo tensor con límites.

Ejemplo 1:

En este ejemplo, veremos cómo rellenar los límites del tensor de entrada con cero.

# Import required library
import torch
import torch.nn as nn
 
# define a tensor
tens = torch.tensor([[[11, 12], [13, 14]]])
print("\n Input Tensor: \n", tens)
 
# give padding size same for all sides
pad = nn.ZeroPad2d(1)
output = pad(tens)
 
# display result
print("\n After Pad Input Tensor: \n", output)

Producción:

 

Ejemplo 2:

En este ejemplo, veremos cómo agregar tamaños de relleno únicos en todos los lados.

# Import required library
import torch
import torch.nn as nn
 
# define a tensor
tens = torch.tensor([[[11, 12], [13, 14]]])
print("\n Input Tensor: \n", tens)
 
# add unique padding sizes to all sides
# (left, right, top, bottom)
pad = nn.ZeroPad2d((1, 2, 3, 4))
output = pad(tens)
 
# display result
print("\n After Pad Input Tensor:\n", output)

Producción: