Функция PyTorch torch.empty_strided
Справочное руководство Pytorch torch
torch.empty_stridedЭто функция в PyTorch для создания неинициализированного тензора с указанным шагом.
Определение функции
torch.empty_strided(size, stride, dtype=None, device=None, requires_grad=False)
Пример использования
Пример
import torch
# Создайте тензор формы (3, 4) с шагом (1, 3)
x = torch.empty_strided((3, 4), (1, 3))
print(x)
print("Форма:", x.shape)
print("Stride:", x.stride())
print("Примечание: значения — это неинициализированные значения памяти")
# Создайте тензор формы (3, 4) с шагом (1, 3)
x = torch.empty_strided((3, 4), (1, 3))
print(x)
print("Форма:", x.shape)
print("Stride:", x.stride())
print("Примечание: значения — это неинициализированные значения памяти")
Другие расширения