0430f0811e0a5cd377b9c377e915bbbbb1572ce7,pycox/models/deephit.py,DeepHit,predict_pmf,#DeepHit#Any#Any#Any#Any#Any#Any#,399

Before Change

        pmf = _pad_phi(preds.view(preds.size(0), -1)).softmax(1)[:, :-1]
        pmf = pmf.view(preds.shape).transpose(0, 1).transpose(1, 2)
        if numpy is None:
            if tuplefy(input).type() is np.ndarray:
                pmf = pmf.cpu().numpy()
        elif numpy is True:
                pmf = pmf.cpu().numpy()
        return pmf

After Change

        preds = self.predict(input, batch_size, False, eval_, False, to_cpu, num_workers)
        pmf = pad_col(preds.view(preds.size(0), -1)).softmax(1)[:, :-1]
        pmf = pmf.view(preds.shape).transpose(0, 1).transpose(1, 2)
        return array_or_tensor(pmf, numpy, input)

Instances