HF RL Explorer

Task: Get Output Dimensions for Each Layer of a Neural Network in PyTorch

Task: Get Output Dimensions for Each Layer of a Neural Network in PyTorch: a task in Terminal-Lego-15k (Harbor dataset). Write a Python script that takes a PyTorch neural network model and prints the output dimension/shape for each layer when given a specific input tensor. This is useful for…

The task

Write a Python script that takes a PyTorch neural network model and prints the output dimension/shape for each layer when given a specific input tensor. This is useful for debugging and understanding the data flow through a neural network.

Part of PrimeIntellect/Terminal-Lego-15k.