[Bug] Allow use kwargs to specify dtype of to method #24
Description
reg = self.register_tensor(...)
reg_fp32 = reg.to(dtype=float32)will trigger an error since we use kwargs to specify the dtype.
{{ message }}
reg = self.register_tensor(...)
reg_fp32 = reg.to(dtype=float32)will trigger an error since we use kwargs to specify the dtype.