PyTorch función torch.range
Pytorch manual de referencia de torch
torch.rangeEs una función en PyTorch que se utiliza para crear un tensor de secuencia aritmética que incluye el valor end.
Definición de la función
torch.range(start=0, end, step=1, dtype=None, device=None, requires_grad=False)
Ejemplo de uso
Ejemplo
import torch
# Crear una secuencia aritmética de 0 a 5 (incluyendo 5)
x = torch.range(5)
print(x)
# Crear una secuencia aritmética de 1 a 10, con paso de 2
y = torch.range(1, 10, 2)
print(y)
# Crear una secuencia aritmética de 0 a 5 (incluyendo 5)
x = torch.range(5)
print(x)
# Crear una secuencia aritmética de 1 a 10, con paso de 2
y = torch.range(1, 10, 2)
print(y)
Otras extensiones