Función torch.ones de PyTorch

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

torch.onesEs una función en PyTorch utilizada para crear tensores de unos. Crea un tensor con una forma especificada, todos sus elementos inicializados en 1.

Esto se usa comúnmente en aprendizaje profundo para inicializar ciertos parámetros específicos, crear máscaras o como valor de referencia para operaciones matemáticas.

Definición de la función

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

Parámetros:

  • *size(int): la forma del tensor, por ejemplo3、(3, 4)、(2, 3, 4)etc.
  • dtype(torch.dtype, opcional): especifica el tipo de datos del tensor, por defecto estorch.float32。
  • device(torch.device, opcional): especifica el dispositivo donde se almacena el tensor.
  • requires_grad(bool, opcional): indica si se requiere calcular gradientes.
  • pin_memory(bool, opcional): indica si usar memoria fija (pinned memory).

Valor de retorno:

  • torch.Tensor: Devuelve un tensor de unos.

Ejemplos de uso

Ejemplo 1: Crear un tensor unidimensional de unos

Ejemplo

import torch

# Crear un tensor de unos con 5 elementos
x = torch.ones(5)

print(x)

El resultado es:

tensor([1., 1., 1., 1., 1.])

Ejemplo 2: Crear un tensor bidimensional de unos

Ejemplo

import torch

# Crear un tensor de unos de 3x4 (matriz)
x = torch.ones(3, 4)

print(x)
print(x.shape)

El resultado es:

tensor([[1., 1., 1., 1.],
        [1., 1., 1., 1.],
        [1., 1., 1., 1.]])
torch.Size([3, 4])

Ejemplo 3: Crear un tensor tridimensional de unos

Ejemplo

import torch

# Crear un tensor de unos de 2x3x4
x = torch.ones(2, 3, 4)

print(x.shape)

El resultado es:

torch.Size([2, 3, 4])

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

Otras extensiones