Function tfc::utils::TFCUtils::zerosRobust

Contents

Function tfc::utils::TFCUtils::zerosRobust#

Defined in File TFCUtils.py

Function Documentation#

tfc.utils.TFCUtils.zerosRobust(PyTree val)#

Returns zeros_like val, but can handle arrays and dictionaries.

Parameters
----------
val : PyTree

Returns
-------
zeros_like_val : PyTree
    Pytree with the same structure as val with all elements equal to zero.