Module border_tch_agent::util

source · [−]
Expand description

Utilities.

Structs

NamedTensors

Named tensors to send model parameters using a channel.

Enums

CriticLoss

Critic loss type.

Traits

OutDim

Returns the dimension of output vectors, i.e., the number of discrete outputs.

Functions

concat_slices

Concatenates slices.

quantile_huber_loss

Returns the quantile huber loss.

track

Apply soft update on a model.