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

Add TODO

Browse source
This commit is contained in:
Aodhnait Étaín 2020-11-15 22:11:05 +00:00
parent f8101400c2
commit de084e8ce3
1 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
pygrad/tensor.py
View file

@ -6,8 +6,7 @@ def tensor(*args, **kwargs):
class Tensor: class Tensor:
# TODO Implement 'requires_grad' functionality. # TODO Implement 'requires_grad' functionality.
def __init__(self, value): def __init__(self, value):
# NOTE We technically could support both numpy arrays and scalar values, # TODO Add support for scalar values.
# but it is too much work.
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)