Type-level max function for use in other type computations
Input tensor storage type
Axes to reduce along
Whether to keep reduced dimensions
type Input = TensorStorage<Float32, [2, 3, 4], [12, 4, 1], DefaultLayoutFlags>;type BatchMax = Max<Input, [0]>; // Max over batch dimension Copy
type Input = TensorStorage<Float32, [2, 3, 4], [12, 4, 1], DefaultLayoutFlags>;type BatchMax = Max<Input, [0]>; // Max over batch dimension
Type-level max function for use in other type computations