Skip to content

README.md

Latest commit

 

History

History
16 lines (11 loc) · 412 Bytes

README.md

File metadata and controls

16 lines (11 loc) · 412 Bytes

Tensor

A basic tensor class (~numpy like)

Usage

from tensor import Tensor

t = Tensor([[1, 2], [3, 4]])
t = Tensor([1, 2, 3, 4], shape=(2, 2))
t = Tensor(shape=(3, 5))

Operations like elt-wise add (+), mult (*), matmul (@), indexing ([]), and reshaping (.reshape()) work (~) similarly to NumPy arrays, with broadcasting and shape infer support

See test_tensor.py for examples.