rlmesh.jax.from_array
Encode a JAX array as an RLMesh value.
from_array(array: object) -> Tensor | PrimitiveValueParameters
| Name | Type | Description |
|---|---|---|
array | object | JAX array to encode. |
Returns
| Type | Description |
|---|---|
Tensor | PrimitiveValue | Tensor for non-scalar arrays, or a primitive for scalar values. |