minor fix

This commit is contained in:
Your Name
2022-07-20 17:23:53 +02:00
committed by Nicola Demo
parent 75a81af99c
commit a05adea4e3
10 changed files with 231 additions and 203 deletions

View File

@@ -119,7 +119,6 @@ class PINN(object):
return self
def span_pts(self, *args, **kwargs):
"""
>>> pinn.span_pts(n=10, mode='grid')
@@ -141,8 +140,11 @@ class PINN(object):
tensor2.repeat_interleave(n1, dim=0),
labels=tensor2.labels)
return tensor1.append(tensor2)
elif len(tensors):
elif len(tensors) == 1:
return tensors[0]
else:
recursive_result = merge_tensors(tensors[1:])
return merge_tensors([tensors[0], recursive_result])
if isinstance(args[0], int) and isinstance(args[1], str):
argument = {}