torch sample: workaround some of double-dispose bugs
This commit is contained in:
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)
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user