Tensor Puzzles is an interactive collection of 21 exercises for learning tensor programming in PyTorch and NumPy. Each puzzle asks the learner to recreate a familiar array operation from first principles. Solutions must fit on one short line and use only a restricted set of indexing, arithmetic, comparison, and broadcasting tools. Standard convenience functions such as sum, view, squeeze, and take are intentionally prohibited. This constraint encourages a deeper understanding of shapes, indexing, and vectorized computation. Automated tests and visual examples help learners check their reasoning as they progress. The notebook is designed for Google Colab and is accompanied by a video walkthrough.

Features

  • Twenty-one tensor programming puzzles
  • PyTorch and NumPy exercises
  • Broadcasting-focused problem solving
  • Restricted one-line solutions
  • Automated tests and visual examples
  • Google Colab and video walkthrough

Project Samples

Project Activity

See All Activity >

Categories

Machine Learning

License

MIT License

Follow Tensor Puzzles

Tensor Puzzles Web Site

Other Useful Business Software
Ship Agents Faster Icon
Ship Agents Faster

Transform your applications and workflows into powerful agentic systems at global scale.

Gemini Enterprise Agent Platform lets you rapidly build, scale, govern and optimize production-ready agents grounded in your organization's data. The platform enables developers to build custom or pre-built agents for virtually any use case. New customers get $300 in free credits.
Get Started Free
Rate This Project
Login To Rate This Project

User Reviews

Be the first to post a review of Tensor Puzzles!

Additional Project Details

Programming Language

Python

Related Categories

Python Machine Learning Software

Registered

4 days ago