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