Skip to content

PyTorch practice problems

Difficulty:
Category & tags Filters active
Category:
Tag:

Showing 24 of 24

Title Difficulty
Apply Along an Axis Medium
Batched Matrix Multiply Medium
Bidirectional RNN Medium
Broadcasting Addition Easy
Causal Attention with Framework Operations Hard
Compute Gradient Medium
Compute Jacobian Hard
Concatenate Tensors Easy
Create a Tensor from a List Easy
Cumulative Sum Easy
Custom Activation with Gradient Hard
Einstein Summation Medium
Element-wise Operations Easy
Gather Elements Medium
Masked Fill Medium
Matrix Multiplication Easy
Mixed Precision Forward Pass Hard
Per-Sample Gradients with vmap Hard
Reshape a Tensor Easy
Scatter Add Medium
Stack Tensors Easy
Tensor Indexing and Slicing Easy
Top-K Values Medium
Transpose a Matrix Easy