copying utilities for tensors

This commit is contained in:
Roland Grinis 2021-07-08 09:43:55 +01:00
parent 40d8a05bb0
commit 8f32397f50

View File

@ -28,6 +28,7 @@ constructor(scope: NoaScope, internal val tensorHandle: TensorHandle) :
get() = (1..dimension).map { JNoa.getStrideAt(tensorHandle, it - 1) }.toIntArray()
public val numElements: Int get() = JNoa.getNumel(tensorHandle)
public val device: Device get() = Device.fromInt(JNoa.getDevice(tensorHandle))
override fun toString(): String = JNoa.tensorToString(tensorHandle)
@ -48,21 +49,21 @@ constructor(scope: NoaScope, internal val tensorHandle: TensorHandle) :
scope = scope,
tensorHandle = JNoa.copyToDouble(this.tensorHandle)
)
/*
public fun copyToFloat(): TorchTensorFloat = TorchTensorFloat(
public fun copyToFloat(): NoaFloatTensor = NoaFloatTensor(
scope = scope,
tensorHandle = JTorch.copyToFloat(this.tensorHandle)
tensorHandle = JNoa.copyToFloat(this.tensorHandle)
)
public fun copyToLong(): TorchTensorLong = TorchTensorLong(
public fun copyToLong(): NoaLongTensor = NoaLongTensor(
scope = scope,
tensorHandle = JTorch.copyToLong(this.tensorHandle)
tensorHandle = JNoa.copyToLong(this.tensorHandle)
)
public fun copyToInt(): TorchTensorInt = TorchTensorInt(
public fun copyToInt(): NoaIntTensor = NoaIntTensor(
scope = scope,
tensorHandle = JTorch.copyToInt(this.tensorHandle)
)*/
tensorHandle = JNoa.copyToInt(this.tensorHandle)
)
}
public sealed class NoaTensorOverField<T>
@ -134,9 +135,6 @@ internal constructor(scope: NoaScope, tensorHandle: TensorHandle) :
}
}
public sealed class Device {
public object CPU : Device() {
override fun toString(): String {