10 lines
164 BLFS
Plaintext
10 lines
164 BLFS
Plaintext
name: jax-reduce
|
|
description: Reduce over axis
|
|
inputs:
|
|
array: jax.Array
|
|
op: string
|
|
axis: int
|
|
outputs:
|
|
array: jax.Array
|
|
implementation: jax_skills.reduce_op
|