Функция PyTorch torch.as_tensor
Справочное руководство Pytorch torch
torch.as_tensorявляется функцией в PyTorch, используемой для преобразования данных в тензоры. В отличие отtorch.tensorона по возможности разделяет память данных.
Определение функции
torch.as_tensor(data, dtype=None, device=None)
Пример использования
Пример
import torch
import numpy as np
# Создание из массива NumPy (общая память)
numpy_array = np.array([1, 2, 3])
x = torch.as_tensor(numpy_array)
# Изменение массива NumPy
numpy_array[0] = 100
print("Tensor:", x) # Тензор также изменится.
import numpy as np
# Создание из массива NumPy (общая память)
numpy_array = np.array([1, 2, 3])
x = torch.as_tensor(numpy_array)
# Изменение массива NumPy
numpy_array[0] = 100
print("Tensor:", x) # Тензор также изменится.
Результат вывода:
张量: tensor([100, 2, 3])
Другие расширения