From b7ea55458d9608ef2d03d39343c725c6d130f94a Mon Sep 17 00:00:00 2001
From: Ryan Nett <rnett@calpoly.edu>
Date: Sun, 15 Nov 2020 20:45:21 -0800
Subject: [PATCH] Add default methods to Operand to get the shape and datatype
 from `asOutput()`.

Signed-off-by: Ryan Nett <rnett@calpoly.edu>
---
 .../src/main/java/org/tensorflow/Operand.java     | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

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<T> 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<T> dataType(){
+    return asOutput().dataType();
+  }
 }