Skip to main content

ones_client

Function ones_client 

Source
pub fn ones_client<R: CubeRuntime>(
    client: ComputeClient<R>,
    device: R::Device,
    shape: Shape,
    dtype: DType,
) -> CubeTensor<R>
Expand description

Creates a tensor filled with ones