torch sample: workaround some of double-dispose bugs

This commit is contained in:
Svyatoslav Scherbina
2020-07-22 19:24:37 +03:00
committed by SvyatoslavScherbina
parent 21e780101a
commit b940c6da7e
@@ -34,7 +34,11 @@ abstract class FloatTensor(val raw: CPointer<THFloatTensor>) : Disposable {
fun sum() = THFloatTensor_sumall(raw)
fun flatten() = (0 until elements.size).map { data[it] }.toTypedArray()
private var disposed = false
override fun dispose() {
if (disposed) return
disposed = true
THFloatTensor_free(raw)
}