nip.utils.tensordict

Utilities for manipulating tensordicts.

Functions

get_key_batch_size(td, key)	Get the batch_size of the inner-most tensordict containing the key.
tensordict_add(td1, td2, *[, inplace])	Add two tensordicts together.
tensordict_scalar_multiply(td, scalar, *[, ...])	Multiply a tensordict by a scalar.