diff --git a/src/TensorFlowNET.Core/Operations/Operation.cs b/src/TensorFlowNET.Core/Operations/Operation.cs index 8af7a422..498db789 100644 --- a/src/TensorFlowNET.Core/Operations/Operation.cs +++ b/src/TensorFlowNET.Core/Operations/Operation.cs @@ -275,6 +275,11 @@ namespace Tensorflow return NodeDef.Parser.ParseFrom(buffer.DangerousMemoryBlock.Stream()); } } + + public TensorShape get_shape(Tensor x) + { + return x.shape; + } /// /// Update the input to this operation at the given index. @@ -323,4 +328,4 @@ namespace Tensorflow return new TF_Input(op, input_idx); } } -} \ No newline at end of file +}