public casting

This commit is contained in:
Roland Grinis 2021-07-09 10:12:30 +01:00
parent 68d0e9958f
commit b8ff5938ff

View File

@ -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
*/ */