def test_convert_tensor_like(self, t1, t2): """Test that converting t1 like t2 results in t1 being cast to the same tensor type as t2""" res = fn.convert_like(t1, t2) # if tensorflow or pytorch, extract view of underlying data if hasattr(res, "numpy"): res = res.numpy() if hasattr(t2, "numpy"): t2 = t2.numpy() assert fn.allequal(res, t1) assert isinstance(res, np.ndarray if isinstance(t2, (list, tuple)) else t2.__class__)
def test_convert_scalar(self, t_like): """Test that a python scalar is converted to a scalar tensor""" res = fn.convert_like(5, t_like) assert isinstance(res, t_like.__class__) assert res.ndim == 0 assert fn.allequal(res, [5])