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?
Thank you! We'll create more content like this.
Thank you for helping us improve!