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