Función torch.ones_like de PyTorch

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

torch.ones_likeEs una función en PyTorch que se utiliza para crear un tensor de todos unos con la misma forma que el tensor de entrada.

Definición de la función

torch.ones_like(input, dtype=None, device=None, requires_grad=False)

Ejemplo de uso

Ejemplo

import torch

x = torch.randn(2, 3)
y = torch.ones_like(x)

print(y)

El resultado de salida es:

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

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

Otras extensiones