forked from kscience/kmath
copying utilities for tensors
This commit is contained in:
parent
40d8a05bb0
commit
8f32397f50
@ -28,6 +28,7 @@ constructor(scope: NoaScope, internal val tensorHandle: TensorHandle) :
|
|||||||
get() = (1..dimension).map { JNoa.getStrideAt(tensorHandle, it - 1) }.toIntArray()
|
get() = (1..dimension).map { JNoa.getStrideAt(tensorHandle, it - 1) }.toIntArray()
|
||||||
|
|
||||||
public val numElements: Int get() = JNoa.getNumel(tensorHandle)
|
public val numElements: Int get() = JNoa.getNumel(tensorHandle)
|
||||||
|
|
||||||
public val device: Device get() = Device.fromInt(JNoa.getDevice(tensorHandle))
|
public val device: Device get() = Device.fromInt(JNoa.getDevice(tensorHandle))
|
||||||
|
|
||||||
override fun toString(): String = JNoa.tensorToString(tensorHandle)
|
override fun toString(): String = JNoa.tensorToString(tensorHandle)
|
||||||
@ -48,21 +49,21 @@ constructor(scope: NoaScope, internal val tensorHandle: TensorHandle) :
|
|||||||
scope = scope,
|
scope = scope,
|
||||||
tensorHandle = JNoa.copyToDouble(this.tensorHandle)
|
tensorHandle = JNoa.copyToDouble(this.tensorHandle)
|
||||||
)
|
)
|
||||||
/*
|
|
||||||
public fun copyToFloat(): TorchTensorFloat = TorchTensorFloat(
|
public fun copyToFloat(): NoaFloatTensor = NoaFloatTensor(
|
||||||
scope = scope,
|
scope = scope,
|
||||||
tensorHandle = JTorch.copyToFloat(this.tensorHandle)
|
tensorHandle = JNoa.copyToFloat(this.tensorHandle)
|
||||||
)
|
)
|
||||||
|
|
||||||
public fun copyToLong(): TorchTensorLong = TorchTensorLong(
|
public fun copyToLong(): NoaLongTensor = NoaLongTensor(
|
||||||
scope = scope,
|
scope = scope,
|
||||||
tensorHandle = JTorch.copyToLong(this.tensorHandle)
|
tensorHandle = JNoa.copyToLong(this.tensorHandle)
|
||||||
)
|
)
|
||||||
|
|
||||||
public fun copyToInt(): TorchTensorInt = TorchTensorInt(
|
public fun copyToInt(): NoaIntTensor = NoaIntTensor(
|
||||||
scope = scope,
|
scope = scope,
|
||||||
tensorHandle = JTorch.copyToInt(this.tensorHandle)
|
tensorHandle = JNoa.copyToInt(this.tensorHandle)
|
||||||
)*/
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
public sealed class NoaTensorOverField<T>
|
public sealed class NoaTensorOverField<T>
|
||||||
@ -134,9 +135,6 @@ internal constructor(scope: NoaScope, tensorHandle: TensorHandle) :
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
public sealed class Device {
|
public sealed class Device {
|
||||||
public object CPU : Device() {
|
public object CPU : Device() {
|
||||||
override fun toString(): String {
|
override fun toString(): String {
|
||||||
|
Loading…
Reference in New Issue
Block a user