Skip to main content
Log in

Mojo function

layer_norm_reshape

layer_norm_reshape[type: DType, rank: Int, //, output_rank: Int](shape: Index[rank, element_type=element_type], buf: NDBuffer[type, rank, origin, shape, strides]) -> NDBuffer[type, output_rank, origin]