KMP library for tensors #300
@ -61,7 +61,7 @@ public open class DoubleTensorAlgebra :
|
|||||||
*
|
*
|
||||||
* @param shape the desired shape for the tensor.
|
* @param shape the desired shape for the tensor.
|
||||||
* @param initializer mapping tensor indices to values.
|
* @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 =
|
public fun produce(shape: IntArray, initializer: (IntArray) -> Double): DoubleTensor =
|
||||||
fromArray(shape,
|
fromArray(shape,
|
||||||
|
Loading…
Reference in New Issue
Block a user