Translating TF or PyTorch or other code to returnn_common

• tf.shape(x)[axis] -> nn.length(axis) (where axis is nn.Dim) (but not necessarily returning a scalar) or nn.dim_value(axis)

Also see the small RETURNN migration guide and RETURNN common conventions.