Add defined() to caffe2::Tensor (pytorch#16125)

Summary: Pull Request resolved: pytorch#16125 Add defined() method to check whether the Tensor is defined. Reviewed By: ezyang Differential Revision: D13719222 fbshipit-source-id: ff8efef2159ed1026bd16acaea40c768a1e20a47
jiaqianjing · Jan 18, 2019 · da578b7 · da578b7
1 parent b9b160d
commit da578b7
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/caffe2/core/blob_test.cc b/caffe2/core/blob_test.cc
@@ -587,6 +587,11 @@ TEST(TensorTest, Tensor64BitDimension) {
   EXPECT_EQ(tensor.numel(), large_number * 100);
 }
 
+TEST(TensorTest, UndefinedTensor) {
+  Tensor x;
+  EXPECT_FALSE(x.defined());
+}
+
 TEST(TensorDeathTest, CannotCastDownLargeDims) {
   int64_t large_number =
       static_cast<int64_t>(std::numeric_limits<int>::max()) + 1;

diff --git a/caffe2/core/tensor.h b/caffe2/core/tensor.h
@@ -373,6 +373,10 @@ class CAFFE2_API Tensor final {
     return impl_;
   }
 
+  bool defined() const {
+    return impl_;
+  }
+
   /**
    * Returns a const raw void* pointer of the underlying storage. mutable_data()
    * or raw_mutable_data() must have been called prior to this function call.