torch.dtype is int in JIT instead of torch.dtype. So, you loss to properties: is_floating_point and is_complex . Replication import torch @torch ...
確定! 回上一頁