add type_as

Author: Difers

python/paddle/base/dygraph/math_op_patch.py

@@ -104,6 +104,9 @@ def astype(self: Tensor, dtype: DTypeLike) -> Tensor:
 
         return _C_ops.cast(self, dtype)
 
+    def type_as(self: Tensor, other: Tensor) -> Tensor:
+        return self.astype(other.dtype)
+
     def _scalar_elementwise_op_(
         var: Tensor, scale: float, bias: float
     ) -> Tensor:
@@ -225,6 +228,7 @@ def _mT_(var: Tensor) -> Tensor:
         ('__len__', _len_),
         ('__index__', _index_),
         ('astype', astype),
+        ('type_as', type_as),
         ('dim', dim),
         ('ndimension', ndimension),
         ('ndim', _ndim),

python/paddle/base/layers/math_op_patch.py

@@ -382,6 +382,9 @@ def astype(self, dtype):
         out.stop_gradient = self.stop_gradient
         return out
 
+    def type_as(self, other):
+        return self.astype(other.dtype)
+
     @static_only
     def append(self, var):
         """
@@ -799,6 +802,7 @@ def to_dense(var):
         ('__neg__', _neg_),
         ('__abs__', _abs_),
         ('astype', astype),
+        ('type_as', type_as),
         ('cpu', cpu),
         ('cuda', cuda),
         ('place', place),

python/paddle/pir/math_op_patch.py

@@ -370,6 +370,9 @@ def astype(self, dtype):
 
         return _C_ops.cast(self, dtype)
 
+    def type_as(self, other):
+        return self.astype(other.dtype)
+
     def _scalar_add_(var, value):
         return paddle.scale(var, 1.0, value)
 
@@ -1109,6 +1112,7 @@ def register_hook(self, hook):
         ('ndimension', ndimension),
         ('ndim', _ndim),
         ('astype', astype),
+        ('type_as', type_as),
         ('size', _size_),
         ('T', _T_),
         ('mT', _mT_),