splitTensor
internal fun splitTensor(tensorToSplit: Array3D<FloatArray>, dimensions: LongArray): Pair<Array3D<FloatArray>, Array3D<FloatArray>>(source)
Executes the splitTensor step in the SDAI ONNX local diffusion feature layer.
Return
Result produced by splitTensor.
Author
Dmitriy Moroz
Parameters
tensorToSplit
tensor to split value consumed by the API.
dimensions
dimensions value consumed by the API.