torch.linspace() en pytorch genera N puntos uniformes en un cierto intervalo

Detalles de la función:

torch.linspace(inicio, final, pasos, salida)

• Función : genera N puntos uniformes en el intervalo [inicio, fin].
• Parámetros :
  • start (float) : el punto de inicio del intervalo;
  • end (float) : el punto final del intervalo;
  • pasos(int) : el número de puntos generados;
  • out (Tensor, opcional) : El tensor de resultado.
• Nota : el inicio puede ser menor que el final, mayor que el final o igual al final.

Ejemplo:

import torch

a = torch.linspace(0, 1, 5)
print(a)  ## tensor([0.0000, 0.2500, 0.5000, 0.7500, 1.0000])

a = torch.linspace(1, 0, 5)
print(a)  ## tensor([1.0000, 0.7500, 0.5000, 0.2500, 0.0000])

a = torch.linspace(1, 1, 5)
print(a)  ## tensor([1., 1., 1., 1., 1.])