
layout: blog_detail
title: 'Feature Extraction in TorchVision using Torch FX'
author: Alexander Soare and Francisco Massa
featured-img: 'assets/images/fx-image2.png'


Introduction
FX based feature extraction is a new TorchVision utility that lets us access intermediate transformations of an input during the forward pass of a PyTorch Module. It does so by symbolically tracing the forward method to produce a graph where each node represents a single operation. Nodes are named in a human-readable manner such that one may easily specify which nodes they want to access.