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]

Returns:

IndexList

Was this page helpful?