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


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

torch.asarrayявляется функцией в PyTorch, используемой для преобразования данных в тензорные массивы.

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

torch.asarray(data, dtype, device)

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

Пример

import torch

# Преобразование из списка
x = torch.asarray([1, 2, 3])
print(x)

# Преобразование из массива NumPy
import numpy as np
arr = np.array([4, 5, 6])
y = torch.asarray(arr)
print(y)

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

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