function
max
max(src: Buffer[type, size, address_space]) -> SIMD[$0, 1]
Computes the max element in a buffer.
Args:
- src (
Buffer[type, size, address_space]
): The buffer.
Returns:
The maximum of the buffer elements.
max[reduce_axis: Int](src: NDBuffer[type, rank, shape, address_space], dst: NDBuffer[type, rank, shape, 0])
Computes the max across reduce_axis of an NDBuffer.
Parameters:
- reduce_axis (
Int
): The axis to reduce across.
Args:
- src (
NDBuffer[type, rank, shape, address_space]
): The input buffer. - dst (
NDBuffer[type, rank, shape, 0]
): The output buffer.