public casting
This commit is contained in:
parent
68d0e9958f
commit
b8ff5938ff
@ -22,6 +22,8 @@ protected constructor(protected val scope: NoaScope) :
|
|||||||
|
|
||||||
protected abstract fun wrap(tensorHandle: TensorHandle): TensorType
|
protected abstract fun wrap(tensorHandle: TensorHandle): TensorType
|
||||||
|
|
||||||
|
public fun Tensor<T>.cast(): TensorType = tensor
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* A scalar tensor must have empty shape
|
* A scalar tensor must have empty shape
|
||||||
*/
|
*/
|
||||||
|
Loading…
Reference in New Issue
Block a user