map method

Tensor map(double fn(double))

Apply a function to each element.

Implementation

Tensor map(double Function(double) fn) {
  final result = Float32List(size);
  for (int i = 0; i < size; i++) {
    result[i] = fn(data[i]);
  }
  return Tensor(result, List<int>.from(shape));
}