Skip to main content

Mojo function

rebuild_static_tensor_specs_with_output_lambda

rebuild_static_tensor_specs_with_output_lambda[func_type: AnyTrivialRegType, //, dtype: DType, rank: Int](spec: StaticTensorSpec[dtype, rank], out_lambda: func_type) -> StaticTensorSpec[dtype, rank]

Was this page helpful?