9 lines
164 BLFS
Plaintext
9 lines
164 BLFS
Plaintext
name: jax-map
|
|
description: Apply elementwise mapping function
|
|
inputs:
|
|
array: jax.Array
|
|
op: string
|
|
outputs:
|
|
array: jax.Array
|
|
implementation: jax_skills.map_op
|