aodhneine/pygrad
1
1
Fork 0
Code Issues Pull requests Releases Wiki Activity

Add support for creating tensors from lists

Browse source
This commit is contained in:
Aodhnait Étaín 2020-11-16 00:44:56 +00:00
parent 12d229e502
commit d31a72374d
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
pygrad/tensor.py
View file

@ -7,6 +7,8 @@ class Tensor:
# TODO Implement 'requires_grad' functionality. # TODO Implement 'requires_grad' functionality.
def __init__(self, value): def __init__(self, value):
# TODO Add support for scalar values. # TODO Add support for scalar values.
if isinstance(value, list):
value = np.array(value)
if not isinstance(value, np.ndarray): if not isinstance(value, np.ndarray):
print(f"{type(value)} is not compatible with {np.ndarray}") print(f"{type(value)} is not compatible with {np.ndarray}")
exit(-1) exit(-1)