Skip to main content

Mojo function

layer_norm_reshape

layer_norm_reshape[rank: Int, //, output_rank: Int](shape: IndexList[rank, element_type=element_type]) -> IndexList[output_rank]

Was this page helpful?