f4ca9b361adb382423c753f7f864a5dd0e454964,torchsample/transforms/tensor_transforms.py,TypeCast,__call__,#TypeCast#,171

Before Change

        self.dtype = dtype

    def __call__(self, x, y=None):
        x = x.type(self.dtype)
        if y is not None:
            y = y.type(self.dtype)
            return x, y
        return x

After Change

        self.dtype = dtype

    def __call__(self, *inputs):
        outputs = []
        for idx, _input in enumerate(inputs):
            _input = _input.type(self.dtype)
            outputs.append(_input)
        return outputs if idx > 1 else outputs[0]

Instances