torchwrench.nn.functional.checksum module¶

torchwrench.nn.functional.checksum.checksum_dataframe(x: DataFrame, **kwargs) → int[source]¶

torchwrench.nn.functional.checksum.checksum_dtype(x: dtype | dtype, **kwargs) → int[source]¶

torchwrench.nn.functional.checksum.checksum_module(x: Module, *, only_trainable: bool = False, with_names: bool = False, buffers: bool = False, training: bool = False, **kwargs) → int[source]¶

Compute a simple checksum over module parameters.

torchwrench.nn.functional.checksum.checksum_numpy(x: ndarray | generic, **kwargs) → int[source]¶

Compute a simple checksum of a tensor. Order of values matter for the checksum.

torchwrench.nn.functional.checksum.checksum_series(x: Series, **kwargs) → int[source]¶

torchwrench.nn.functional.checksum.checksum_tensor(x: Tensor, **kwargs) → int[source]¶

Compute a simple checksum of a tensor. Order of values matter for the checksum.