(s, name, path)
| 4 | from tinygrad.tensor import Tensor |
| 5 | |
| 6 | def save_scalar(s, name, path): |
| 7 | s = np.array([1.0, float(s)]).astype(np.float32) |
| 8 | np.save(pathlib.Path(path, f'{name}.npy'), s) |
| 9 | |
| 10 | def save_tensor(tensor, name, path): |
| 11 | tensor_numpy = tensor.numpy() |
no outgoing calls
no test coverage detected