promptdojo_

Tensors and shapes before anything else — step 2 of 7

x has shape (32, 10) — a batch of 32 rows, 10 features each. What shape does x.mean(dim=0) return, and what does it mean?