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
|
||||
|
||||
public fun Tensor<T>.cast(): TensorType = tensor
|
||||
|
||||
/**
|
||||
* A scalar tensor must have empty shape
|
||||
*/
|
||||
|
Loading…
Reference in New Issue
Block a user