Language

unilab.algos.torch.flash_sac.learner.RunningMeanStd¶

class unilab.algos.torch.flash_sac.learner.RunningMeanStd[source]¶

Bases: object

RunningMeanStd(mean: ‘torch.Tensor’, var: ‘torch.Tensor’, count: ‘torch.Tensor’)

Parameters:

mean (Tensor)
var (Tensor)
count (Tensor)

Methods

`__init__`(mean, var, count)
`create`(device)
`load_state_dict`(state_dict)
`state_dict`()
`update`(x)

Attributes

`mean`
`var`
`count`

mean: Tensor¶

var: Tensor¶

count: Tensor¶

classmethod create(device)[source]¶

Parameters:: device (device)
Return type:: RunningMeanStd

update(x)[source]¶

Parameters:: x (Tensor)
Return type:: None

state_dict()[source]¶

Return type:: dict[str, Tensor]

load_state_dict(state_dict)[source]¶

Parameters:: state_dict (dict[str, Tensor])
Return type:: None

__init__(mean, var, count)¶

Parameters:

mean (Tensor)
var (Tensor)
count (Tensor)