/// <summary>
/// Returns the shape of the Tensor
/// </summary>
/// <param name="output">The output</param>
/// <param name="numDim">The rank of the tensor</param>
/// <param name="status">The status</param>
/// <returns>The shape of the Tensor</returns>
public int[] GetTensorShape(Output output, int numDim, Status status = null)
{
int[] dims = new int[numDim];
GCHandle handle = GCHandle.Alloc(dims, GCHandleType.Pinned);
using (StatusChecker checker = new StatusChecker(status))
TfInvoke.tfeGraphGetTensorShape(_ptr, output.Operation, output.Index, handle.AddrOfPinnedObject(), numDim, checker.Status);
handle.Free();
return(dims);
}
