tch-rs icon indicating copy to clipboard operation
tch-rs copied to clipboard

Published 20 hours ago • verified publisher icon LaurentMazare

Perform operations on `Tensor` lazily?

Open NobodyXu opened this issue 1 year ago • 2 comments

Is there anyway to perform operations on Tensor lazily and records operations into the graph to maximize the performance like this one?

NobodyXu avatar Sep 07 '22 09:09 NobodyXu

I don't think there is any support for LazyTensor or similar at the moment (and looking at your link, note that there is no support for an XLA backend neither).

LaurentMazare avatar Oct 09 '22 08:10 LaurentMazare

Thanks!

NobodyXu avatar Oct 09 '22 12:10 NobodyXu