kscience/kmath
9
0
Fork 1
Code Issues Pull Requests 6 Packages Projects Releases 16 Wiki Activity

copying utilities for tensors

Browse Source
This commit is contained in:
Roland Grinis 2021-07-08 09:43:55 +01:00
parent 40d8a05bb0
commit 8f32397f50
1 changed files with 9 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
kmath-noa/src/main/kotlin/space/kscience/kmath/noa/tensors.kt
View File

@ -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 {