Función torch.tile de PyTorch

Pytorch torch 参考手册Manual de referencia de PyTorch torch

torch.tileEs la función en PyTorch utilizada para repetir tensores varias veces.

Definición de la función

torch.tile(input, dims)

Ejemplo de uso

Ejemplo

import torch

# Repetir tensor unidimensional
x = torch.tensor([1, 2, 3])
print("Tensor unidimensional original:")
print(x)

result = torch.tile(x, 3)
print("Repetir 3 veces:")
print(result)

# Repetir tensor bidimensional
y = torch.tensor([[1, 2], [3, 4]])
print("nTensor bidimensional original:")
print(y)

result = torch.tile(y, (2, 3))
print("Repetir 2 veces a lo largo de las filas, 3 veces a lo largo de las columnas:")
print(result)

# Repetir tensor tridimensional
z = torch.tensor([[[1, 2], [3, 4]]])
print("nTensor tridimensional original:")
print(z)

result = torch.tile(z, (2, 1, 2))
print("Repetir forma [2, 1, 2]:")
print(result)
print(f"Forma: {result.shape}")

El resultado es:

原始一维张量:
tensor([1, 2, 3])
重复 3 次:
tensor([1, 2, 3, 1, 2, 3, 1, 2, 3])

原始二维张量:
tensor([[1, 2],
        [3, 4]])
沿行重复 2 次，沿列重复 3 次:
tensor([[1, 2, 1, 2, 1, 2],
        [3, 4, 3, 4, 3, 4],
        [1, 2, 1, 2, 1, 2],
        [3, 4, 3, 4, 3, 4]])

原始三维张量:
tensor([[[1, 2],
         [3, 4]]])
形状 [2, 1, 2] 重复:
tensor([[[1, 2, 1, 2],
         [3, 4, 3, 4]],

        [[1, 2, 1, 2],
         [3, 4, 3, 4]]])
形状: torch.Size([2, 2, 4])

Pytorch torch 参考手册Manual de referencia de PyTorch torch

Otras extensiones