search

Bihaqo / t3f

Tensor Train decomposition on TensorFlow
https://t3f.readthedocs.io/en/latest/index.html
MIT License
219 stars 55 forks source link

Implement working with batches as a decorator #170

Open Bihaqo opened 5 years ago

Bihaqo commented 5 years ago

Something like

@support_batches
def matmul(a, b):
  ...
matmul(a, b)
matmul(batch_a, b)
matmul(batch_a, batch_b)