This commit is contained in:
Aodhnait Étaín 2020-11-15 22:11:05 +00:00
parent f8101400c2
commit de084e8ce3

View file

@ -6,8 +6,7 @@ def tensor(*args, **kwargs):
class Tensor:
# TODO Implement 'requires_grad' functionality.
def __init__(self, value):
# NOTE We technically could support both numpy arrays and scalar values,
# but it is too much work.
# TODO Add support for scalar values.
if not isinstance(value, np.ndarray):
print(f"{type(value)} is not compatible with {np.ndarray}")
exit(-1)