Функция PyTorch torch.randn_like

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.randn_likeЭто функция в PyTorch, используемая для создания случайного тензора со стандартным нормальным распределением и той же формой, что и входной тензор.

Определение функции

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

Пример использования

Пример

import torch

# Создайте входной тензор
input = torch.tensor([[1, 2], [3, 4]])

# Создать случайный тензор со стандартным нормальным распределением и такой же формой, как у входных данных
x = torch.randn_like(input)

print(x)

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения