Función PyTorch torch.randint_like
Manual de referencia de PyTorch torch
torch.randint_likeEs una función en PyTorch que se utiliza para crear un tensor aleatorio de enteros con la misma forma que el tensor de entrada.
Definición de la función
torch.randint_like(input, low=0, high, dtype=None, device=None, requires_grad=False)
Ejemplo de uso
Ejemplo
import torch
# Crear un tensor de entrada
input = torch.tensor([[1, 2], [3, 4]])
# Crear un tensor aleatorio de enteros con la misma forma que la entrada, rango [0, 10)
x = torch.randint_like(input, 10)
print(x)
# Crear un tensor de entrada
input = torch.tensor([[1, 2], [3, 4]])
# Crear un tensor aleatorio de enteros con la misma forma que la entrada, rango [0, 10)
x = torch.randint_like(input, 10)
print(x)
Otras extensiones