tiny.accy.tensor.trace.value

1pub const value = @import("value.zig");

1const tensor = @import("../root.zig");2const trace = @import("root.zig");34const Builder = trace.Builder;5const CompareDirection = trace.CompareDirection;6const Dim = trace.Dim;7const Id = trace.Id;8const Reducer = trace.Reducer;9const Type = trace.Type;10const builder_mod = @import("builder.zig");11const type_mod = tensor.types;1213pub const Value = struct {14    builder: *Builder,15    id: Id,16    ty: Type,1718    pub fn add(self: Value, rhs: Value) !Value {19        return self.builder.alignedBinary(.add, self, rhs);20    }2122    pub fn sub(self: Value, rhs: Value) !Value {23        return self.builder.alignedBinary(.sub, self, rhs);24    }2526    pub fn mul(self: Value, rhs: Value) !Value {27        return self.builder.alignedBinary(.mul, self, rhs);28    }2930    pub fn div(self: Value, rhs: Value) !Value {31        return self.builder.alignedBinary(.div, self, rhs);32    }3334    pub fn pow(self: Value, rhs: Value) !Value {35        return self.builder.alignedBinary(.pow, self, rhs);36    }3738    pub fn max(self: Value, other: anytype) !Value {39        if (comptime @TypeOf(other) == Value) {40            return self.builder.alignedBinary(.max, self, other);41        }42        return self.builder.reduceNamed(self, .max, valueAxes(other));43    }4445    pub fn min(self: Value, other: anytype) !Value {46        if (comptime @TypeOf(other) == Value) {47            return self.builder.alignedBinary(.min, self, other);48        }49        return self.builder.reduceNamed(self, .min, valueAxes(other));50    }5152    pub fn sum(self: Value, axes: anytype) !Value {53        return self.builder.reduceNamed(self, .sum, valueAxes(axes));54    }5556    pub fn mean(self: Value, axes: anytype) !Value {57        return self.builder.meanNamed(self, valueAxes(axes));58    }5960    pub fn reduce(self: Value, init: Value, reducer: Reducer, axes: anytype) !Value {61        return self.builder.reduceWith(self, init, reducer, valueAxes(axes));62    }6364    pub fn gather(self: Value, indices: Value, comptime axis: anytype) !Value {65        return self.builder.gather(self, indices, axis);66    }6768    pub fn scatterAdd(self: Value, indices: Value, updates: Value, comptime axis: anytype) !Value {69        return self.builder.scatterAdd(self, indices, updates, axis);70    }7172    pub fn sparseCrossEntropyLoss(self: Value, targets: Value, comptime axis: anytype) !Value {73        return self.builder.sparseCrossEntropyLoss(self, targets, axis);74    }7576    pub fn contract(self: Value, rhs: Value, axes: anytype) !Value {77        return self.builder.contract(self, rhs, valueAxes(axes));78    }7980    pub fn rename(self: Value, comptime old_name: anytype, comptime new_name: anytype) !Value {81        return self.builder.renameAxis(self, comptime builder_mod.nameOf(old_name), comptime builder_mod.nameOf(new_name));82    }8384    pub fn split(self: Value, comptime axis: anytype, parts_struct: anytype) !Value {85        var buffer: [type_mod.dimCount(@TypeOf(parts_struct))]Dim = undefined;86        type_mod.fillDims(parts_struct, &buffer);87        return self.builder.splitAxis(self, comptime builder_mod.nameOf(axis), &buffer);88    }8990    pub fn merge(self: Value, axes: anytype, comptime merged_name: anytype) !Value {91        return self.builder.mergeAxes(self, valueAxes(axes), comptime builder_mod.nameOf(merged_name));92    }9394    pub fn broadcast(self: Value, added_struct: anytype) !Value {95        var buffer: [type_mod.dimCount(@TypeOf(added_struct))]Dim = undefined;96        type_mod.fillDims(added_struct, &buffer);97        return self.builder.broadcastAxes(self, &buffer);98    }99100    pub fn isStructuralZero(self: Value) bool {101        return self.builder.isStructuralZero(self.id);102    }103104    pub fn neg(self: Value) !Value {105        return self.builder.unary(.neg, self);106    }107108    pub fn abs(self: Value) !Value {109        return self.builder.unary(.abs, self);110    }111112    pub fn exp(self: Value) !Value {113        return self.builder.unary(.exp, self);114    }115116    pub fn log(self: Value) !Value {117        return self.builder.unary(.log, self);118    }119120    pub fn sqrt(self: Value) !Value {121        return self.builder.unary(.sqrt, self);122    }123124    pub fn tanh(self: Value) !Value {125        return self.builder.unary(.tanh, self);126    }127128    pub fn sin(self: Value) !Value {129        return self.builder.unary(.sin, self);130    }131132    pub fn cos(self: Value) !Value {133        return self.builder.unary(.cos, self);134    }135136    pub fn tan(self: Value) !Value {137        return self.builder.unary(.tan, self);138    }139140    pub fn compare(self: Value, direction: CompareDirection, rhs: Value) !Value {141        return self.builder.alignedCompare(direction, self, rhs);142    }143144    pub fn select(self: Value, on_true: Value, on_false: Value) !Value {145        return self.builder.alignedSelect(self, on_true, on_false);146    }147};148149fn valueAxes(axes: anytype) []const []const u8 {150    const Axes = @TypeOf(axes);151    if (comptime type_mod.isNameSlice(Axes)) {152        return axes;153    } else {154        return comptime type_mod.axisNames(axes);155    }156}