feat(wrapper/tensor): Drop method

example/tensor1/main.go

@@ -62,4 +62,8 @@ func main() {
 	wrapper.MustCopy_(dst, ts)
 	dst.Print()
 
+	ts.MustDrop()
+	// The below statement will be panic as `ts` has been dropped.
+	// ts.Print()
+
 }

libtch/tensor.go

@@ -305,3 +305,9 @@ func AtToString(ts Ctensor, lineSize int64) string {
 
 	return goString
 }
+
+// void at_free(tensor);
+func AtFree(ts Ctensor) {
+	ctensor := (C.tensor)(ts)
+	C.at_free(ctensor)
+}

wrapper/tensor.go

@@ -814,3 +814,20 @@ func (ts Tensor) MustToString(lw int64) (retVal string) {
 
 	return retVal
 }
+
+// Drop drops (frees) the tensor
+func (ts Tensor) Drop() (err error) {
+	lib.AtFree(ts.ctensor)
+	if err = TorchErr(); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+// MustDrop drops the tensor. It will be panic if error
+func (ts Tensor) MustDrop() {
+	if err := ts.Drop(); err != nil {
+		log.Fatal(err)
+	}
+}