PyTorch torch.rand_like función
Manual de referencia de PyTorch torch
torch.rand_likeEs una función de PyTorch utilizada para crear un tensor aleatorio con distribución uniforme y la misma forma que el tensor de entrada.
Definición de la función
torch.rand_like(input, 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 con distribución uniforme [0, 1) de la misma forma que la entrada
x = torch.rand_like(input)
print(x)
# Crear un tensor de entrada
input = torch.tensor([[1, 2], [3, 4]])
# Crear un tensor aleatorio con distribución uniforme [0, 1) de la misma forma que la entrada
x = torch.rand_like(input)
print(x)
Otras extensiones