On this page
torch.result_type
torch.result_type(tensor1, tensor2) → dtype
-
Returns the
torch.dtype
that would result from performing an arithmetic operation on the provided input tensors. See type promotion documentation for more information on the type promotion logic.- Parameters
Example:
>>> torch.result_type(torch.tensor([1, 2], dtype=torch.int), 1.0) torch.float32 >>> torch.result_type(torch.tensor([1, 2], dtype=torch.uint8), torch.tensor(1)) torch.uint8
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.result_type.html