blob: cb0615693aeecaf2b8d7364b3d79b5e80765d0c3 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
package com.yahoo.tensor.functions;
import java.util.function.DoubleBinaryOperator;
import java.util.function.DoubleUnaryOperator;
/**
* The join tensor function.
*
* @author bratseth
*/
public class Map extends PrimitiveTensorFunction {
private final TensorFunction argument;
private final DoubleUnaryOperator mapper;
public Map(TensorFunction argument, DoubleUnaryOperator mapper) {
this.argument = argument;
this.mapper = mapper;
}
public TensorFunction argument() { return argument; }
public DoubleUnaryOperator mapper() { return mapper; }
@Override
public PrimitiveTensorFunction toPrimitive() {
return new Map(argument.toPrimitive(), mapper);
}
}
|