Hashing torch

hash.py

import torch
import io
import numpy as np
import hashlib

x = torch.randn(1, 20)
buff = io.BytesIO()
torch.save(x, buff)
buff.seek(0)

out = hashlib.sha256(buff.getvalue()).hexdigest()
print(out)

def sha256(b):
    if isinstance(b, (int, list, float)):
        b = str(b)
    if isinstance(b, torch.Tensor):
        b = b.cpu().numpy()
    if isinstance(b, np.ndarray):
        b = b.tobytes()
    if type(b) == str:
        b = b.encode()
    if type(b) == bytes:
        return hashlib.sha256(b).hexdigest()
    else:
        raise Exception("Not implemented a method to handle {0}".format(type(b)))
    

print(sha256(x))