diff --git a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/Operand.java b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/Operand.java index fa21f32d4ce..2186b0b4362 100644 --- a/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/Operand.java +++ b/tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/Operand.java @@ -15,6 +15,7 @@ package org.tensorflow; +import org.tensorflow.ndarray.Shape; import org.tensorflow.op.Op; import org.tensorflow.types.family.TType; @@ -76,4 +77,18 @@ default Tensor asTensor() { default T data() { return asOutput().tensor().data(); } + + /** + * Returns the build-time shape of this operand. May be only partially known. + */ + default Shape shape(){ + return asOutput().shape(); + } + + /** + * Returns the {@link DataType} of this operand. + */ + default DataType dataType(){ + return asOutput().dataType(); + } }