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()
|
||||
|
||||
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 {
|
||||
|
Loading…
Reference in New Issue
Block a user