(feat) add device access method

[ccaven]

Returns a reference to the tensor's DeviceStorage

Motivation: create new tensors in forward pass

Example:

fn try_forward(&self, input: ...) {
    let new_tensor: Tensor<Rank2<10, 5>, E, D> = input.device().zeros();
    ....
}