/// <summary> /// Sets the shape of the Tensor referenced by <paramref name="output"/> in graph to /// the shape described by <paramref name="dims"/>. /// </summary> /// <param name="output">The output</param> /// <param name="dims">The shape</param> /// <param name="status">The status</param> public void SetTensorShape(Output output, int[] dims, Status status = null) { GCHandle handle = GCHandle.Alloc(dims, GCHandleType.Pinned); using (StatusChecker checker = new StatusChecker(status)) TfInvoke.tfeGraphSetTensorShape(_ptr, output.Operation, output.Index, handle.AddrOfPinnedObject(), dims.Length, checker.Status); }