mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-06-24 02:14:17 +00:00
9 lines
154 B
Python
Executable file
9 lines
154 B
Python
Executable file
#!/usr/bin/env python3
|
|
import numpy as np
|
|
from tinygrad.tensor import Tensor
|
|
|
|
a = Tensor([-2,-1,0,1,2]).ane()
|
|
print(a.cpu())
|
|
b = a.relu()
|
|
print(b.cpu())
|
|
|