arrow

Serialize tensors in PyTorch 0.4 (#2033)

As of PyTorch 0.4.0, `type(t)` where `t` is a tensor will always equal `torch.Tensor`. To get the original types, you need to call `t.type()`. This change fixes serialization for torch tensors. Otherwise, they're truncated (at least upon deserialization in Ray).