xpag.agents.flax_agents.sac.sac_from_jaxrl.Batch

class Batch(observations, actions, rewards, masks, next_observations)

Bases: tuple

Create new instance of Batch(observations, actions, rewards, masks, next_observations)

Methods

count	Return number of occurrences of value.
index	Return first index of value.

Attributes

actions	Alias for field number 1
masks	Alias for field number 3
next_observations	Alias for field number 4
observations	Alias for field number 0
rewards	Alias for field number 2

__add__(value, /)

Return self+value.

__mul__(value, /)

Return self*value.

actions

Alias for field number 1

count(value, /)

Return number of occurrences of value.

index(value, start=0, stop=9223372036854775807, /)

Return first index of value.

Raises ValueError if the value is not present.

masks

Alias for field number 3

next_observations

Alias for field number 4

observations

Alias for field number 0

rewards

Alias for field number 2