torch sample: workaround some of double-dispose bugs

2020-07-22 19:24:37 +03:00
parent 21e780101a
commit b940c6da7e
1 changed files with 4 additions and 0 deletions
@@ -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)
    }