KMP library for tensors #300
@ -61,7 +61,7 @@ public open class DoubleTensorAlgebra :
|
||||
*
|
||||
* @param shape the desired shape for the tensor.
|
||||
* @param initializer mapping tensor indices to values.
|
||||
* @return tensor with the [shape] shape and data generated by initializer.
|
||||
* @return tensor with the [shape] shape and data generated by the [initializer].
|
||||
*/
|
||||
public fun produce(shape: IntArray, initializer: (IntArray) -> Double): DoubleTensor =
|
||||
fromArray(shape,
|
||||
|
Loading…
Reference in New Issue
Block a user