Skip to main content

ones

Function ones 

Source
pub fn ones<R>(
    device: <R as Runtime>::Device,
    shape: Shape,
    dtype: DType,
) -> RudaTensor<R>
where R: Runtime,
Expand description

Creates a tensor filled with ones