24 lines
746 B
Python
24 lines
746 B
Python
|
import torch
|
||
|
|
|
||
|
|
def to_torch(custom_tensor):
|
||
|
|
shape = custom_tensor.shape
|
||
|
|
pytorch_tensor = torch.zeros(shape)
|
||
|
|
|
||
|
|
def _iterate_indices(shape):
|
||
|
|
if len(shape) == 0:
|
||
|
|
yield ()
|
||
|
|
else:
|
||
|
|
for index in range(shape[0]):
|
||
|
|
for sub_indices in _iterate_indices(shape[1:]):
|
||
|
|
yield (index,) + sub_indices
|
||
|
|
|
||
|
|
# Iterate over all elements using the custom tensor's __getitem__ method
|
||
|
|
for indices in _iterate_indices(shape):
|
||
|
|
value = custom_tensor[indices]
|
||
|
|
pytorch_tensor[tuple(indices)] = value
|
||
|
|
|
||
|
|
return pytorch_tensor
|
||
|
|
|
||
|
|
def compare_torch(tensor1, tensor2, epsilon=1e-5):
|
||
|
|
diff = torch.abs(tensor1 - tensor2)
|
||
|
|
return torch.all(diff < epsilon)
|