PyTorch - Tensor

Tensors are the fundamental building block of machine learning.

Introduction

Tensors are the fundamental building block of machine learning.

Their job is to represent data numerically.

For example, you could represent an image as a tensor with shape [3, 224, 224] which would mean [colour_channels, height, width], as in the image has 3 colour channels (red, green, blue), a height of 224 pixels and a width of 224 pixels.

Project

uv init tensor
cd tensor
uv add torch torchvision

PyTorch would be installed from PyPI, which hosts CPU-only wheels for Windows and macOS, and GPU-accelerated wheels on Linux (targeting CUDA 12.8, as of PyTorch 2.9.1)

This is a valid configuration for projects that want to use CPU builds on Windows and macOS, and CUDA-enabled builds on Linux. However, if you need to support different platforms or accelerators, you’ll need to configure the project accordingly.

Using uv with PYTorch

Pending

Learn PyTorch