arrayd_to_tensor

border_tch_agent::util

Function arrayd_to_tensor 

Source 
pub fn arrayd_to_tensor<T1, T2>(a: ArrayD<T1>, add_batch_dim: bool) -> Tensor
where
    T1: AsPrimitive<T2>,
    T2: Copy + 'static + Element,
Expand description

Converts ndarray::ArrayD to Tensor.