Function read_tensor_atomic_checked 

pub fn read_tensor_atomic_checked<C: Scalar>(
    tensor: Tensor<Atomic<C>>,
    index: usize,
    unroll_factor: usize,
) -> Atomic<C>

Returns the value at index in tensor within bounds.