Files
candle/candle-pyo3/test.py
2023-07-02 20:42:55 +01:00

16 lines
232 B
Python

import candle
t = candle.Tensor(42.0)
print(t)
print(t.shape, t.rank, t.device)
print(t + t)
t = candle.Tensor([3.0, 1, 4, 1, 5, 9, 2, 6])
print(t)
print(t+t)
t = t.reshape([2, 4])
print(t.matmul(t.t()))
print(t.to_dtype("u8"))