Skip to main content

reshape

ruda_tensor_device::dispatch

Function reshape 

Source 
pub fn reshape<R>(tensor: RudaTensor<R>, shape: Shape) -> RudaTensor<R>
where
    R: Runtime,
Expand description

Reshape a jit tensor to a new shape