(arrays)
| 63 | |
| 64 | |
| 65 | def convert_to_tensor(arrays): |
| 66 | if isinstance(arrays, np.ndarray): |
| 67 | v = torch.from_numpy(arrays).float() |
| 68 | ret = v |
| 69 | elif isinstance(arrays, torch.Tensor): |
| 70 | ret = arrays |
| 71 | elif isinstance(arrays, list): |
| 72 | v = torch.from_numpy(np.array(arrays)).float() |
| 73 | elif type(arrays) is dict: |
| 74 | ret = {} |
| 75 | for k, v in arrays.items(): |
| 76 | if isinstance(v, np.ndarray): |
| 77 | v = torch.from_numpy(v).float() |
| 78 | if type(v) is dict: |
| 79 | v = convert_to_tensor(v) |
| 80 | ret[k] = v |
| 81 | return ret |
| 82 | |
| 83 | def convert_like(inp, target): |
| 84 | if isinstance(target, np.ndarray): |
no outgoing calls
no test coverage detected