Función PyTorch torch.rand

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

torch.randEs una función en PyTorch para crear tensores aleatorios a partir de una distribución uniforme [0, 1).

Esto se usa comúnmente en el aprendizaje profundo para inicialización, generación de máscaras aleatorias, etc.

Definición de la función

torch.rand(*size, dtype=None, device=None, requires_grad=False)

Parámetros:

  • *size(int): La forma del tensor.
  • dtype(torch.dtype, opcional): Tipo de dato.
  • device(torch.device, opcional): Dispositivo.
  • requires_grad(bool, opcional): Si se requiere calcular gradientes.

Valor de retorno:

  • torch.Tensor: Devuelve un tensor que contiene números aleatorios, en el rango [0, 1).

Ejemplo de uso

Ejemplo 1: Crear un tensor aleatorio

Ejemplo

import torch

# Crear un tensor aleatorio de 3x4
x = torch.rand(3, 4)

print(x)
print("Rango: [{:.4f}, {:.4f})".format(x.min().item(), x.max().item()))

El resultado de salida es:

tensor([[0.6272, 0.1802, 0.8336, 0.3900],
        [0.2188, 0.6593, 0.3598, 0.4021],
        [0.9247, 0.0144, 0.2587, 0.6634]])
范围: [0.0144, 0.9247)

Ejemplo 2: Crear una máscara aleatoria

Ejemplo

import torch

# Crear una máscara aleatoria con probabilidad del 50%
mask = torch.rand(10) > 0.5

print("Máscara:", mask)
print("Número de True:", mask.sum().item())

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

Otras extensiones