Функция torch.rand_like в PyTorch

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

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

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

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

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

Пример

import torch

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

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

print(x)

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

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