rewritten jnp.squeeze(self._value_layer(inputs), axis=-1)#296
Open
skywalker0803r wants to merge 3 commits into
Open
rewritten jnp.squeeze(self._value_layer(inputs), axis=-1)#296skywalker0803r wants to merge 3 commits into
skywalker0803r wants to merge 3 commits into