nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/autodiff/samediff/SameDiff.java
public long[] getShapeForVarName(String varName) {
        if (variableNameToArr.containsKey(varName)) {
            return variableNameToArr.get(varName).shape();
        }

        return variableNameToShape.get(varName);
    }
public long[] getShapeForVarName(String varName) {
        if (variableNameToArr.containsKey(varName)) {
            return variableNameToArr.get(varName).shape();
        }

        return variableNameToShape.get(varName);
    }
Get the shape for the given vertex id.
Note that if an array is defined, it will use that shape instead.
<p>
A shape *and* an array should not be defined at the same time.
This wastes memory. The internal map used for tracking shapes for particular
vertex ids should also delete redundant shapes stored to avoid redundant sources of information.

@param varName the vertex id to get the shape for
@return the shape for the given vertex if if any.