Тензор (torch.Tensor) — многомерный массив с dtype и device, базовая единица данных для вычислений и хранения параметров модели.
Разбор
- Похож на NumPy ndarray, но умеет жить на GPU и участвовать в autograd.
- Важны
shape, dtype, device и флаг requires_grad.
- Скаляр — 0-D тензор; вектор/матрица — 1-D/2-D.
- Операции обычно elementwise или через BLAS/cuDNN на device.
Пример
x = torch.tensor([[1.0, 2.0], [3.0, 4.0]])
print(x.shape, x.dtype, x.device)
Итог
Тензор — контейнер чисел с метаданными для обучения на CPU/GPU.