View on GitHub
Install in Dash
TensorUtils Class Reference
TensorUtils
public class TensorUtilsUndocumented
-
Randomlly generate a
Tensorobject.Declaration
Swift
public static func generateRandomTensor(targetTensorShape shape: TensorShape, maxVal: Float) -> Tensor -
For a
swiftn-d array, flat it into a 1-d array.array: The given raw array data. We don’t preassume it’s shape.
Note
The returned array could be empty if all elements of the passed in array are not supported types.
Declaration
Swift
public static func flattenArray(array: [Any], dataType type: TensorDataType) -> [Float]Return Value
An array of
SupportedTensorInitialDataTypeelements. It could be empty.