Transforms a numpy.ndarray to torch.Tensor.
| 156 | |
| 157 | |
| 158 | class NumpyToTensor(object): |
| 159 | """ |
| 160 | Transforms a numpy.ndarray to torch.Tensor. |
| 161 | """ |
| 162 | def __call__(self, sample): |
| 163 | return torch.from_numpy(sample) |
no outgoing calls
no test coverage detected
searching dependent graphs…