A Static Analyzer for Detecting Tensor Shape Errors in Deep Neural Network Training Code

Ho Young Jhoo, Sehoon Kim, Woosung Song +3

We present an automatic static analyzer PyTea that detects tensor-shape errors in PyTorch code. The tensor-shape error is critical in the deep neural net code; much of the training…