https://github.com/rballester/tntorch
Tip revision: 3af563a42794ba169e7902198d1edd919617a958 authored by Rafael Ballester on 16 March 2023, 15:48:54 UTC
Updated doc (ranks_cp actually must be an integer, not a list)
Updated doc (ranks_cp actually must be an integer, not a list)
Tip revision: 3af563a
test_init.py
import numpy as np
import tntorch as tn
import torch
torch.set_default_dtype(torch.float64)
def test_from_ndarray():
for i in range(100):
gt = np.random.rand(*np.random.randint(1, 8, np.random.randint(1, 6)))
t = tn.Tensor(gt)
reco = t.numpy()
assert np.linalg.norm(gt - reco) / np.linalg.norm(gt) <+ 1e-7
