jax.ref.freeze

jax.ref.freeze(ref)

使给定的引用失效并返回其最终值。

有关可变数组引用的更多信息，请参阅 Ref 指南。

参数:

ref (Ref) – 一个 jax.ref.Ref 对象。

返回:

一个包含 ref 内容的 jax.Array。

返回类型:

Array

示例

>>> import jax
>>> ref = jax.new_ref(jax.numpy.arange(5))
>>> ref[3] = 100
>>> ref
Ref([  0,   1,   2, 100,   4], dtype=int32)

>>> jax.ref.freeze(ref)
Array([  0,   1,   2, 100,   4], dtype=int32)