vectorLog function
Implementation
Tensor<Vector> vectorLog(Tensor<Vector> v) {
int N = v.data.length;
Vector outValue = [];
for (int i = 0; i < N; i = i + 1) {
outValue.add(log(v.data[i]));
}
Tensor<Vector> out = Tensor<Vector>(outValue);
out.creator = Node(
[v],
() {
for (int i = 0; i < N; i = i + 1) {
v.grad[i] = v.grad[i] + out.grad[i] * (1.0 / v.data[i]);
}
},
opName: 'log_vector',
cost: N,
);
return out;
}