Mojo function
get_batched_scale_factor
get_batched_scale_factor[scales_dtype: DType, scales_layout: Layout, //, SF_VECTOR_SIZE: Int](scales_tensor: LayoutTensor[scales_dtype, scales_layout, MutAnyOrigin], batch_idx: Int, row_idx: Int, col_idx: Int) -> Scalar[scales_dtype]
Returns:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!