tensor.py

#

main.py | graph.py | tensor.py | ops.py | session.py

Previous: The Graph | Next: Operations

from __future__ import absolute_import
from __future__ import print_function
from __future__ import division

import numpy as np
#

Tensor represents a value in the graph. It's just a data container with methods for operator overloading (each of which delegate to the graph). It includes:

  • The initial value of the tensor.
  • The operation which produced the tensor, if applicable.
  • A reference to the graph this tensor belongs to.
class Tensor(object):
# 













#







    def __init__(self, initial_value, op, graph):
        self.initial_value = initial_value
        self.graph = graph
        self.op = op













#



Operator Overloading






    def __add__(self, other):
        return self.graph.add(self, other)













#







    def __sub__(self, other):
        return self.graph.sub(self, other)













#







    def __mul__(self, other):
        return self.graph.mul(self, other)













#







    def __truediv__(self, other):
        return self.graph.div(self, other)













#







    def __neg__(self):
        return self.graph.neg(self)













#



Reverse Operator Overloading






    def __radd__(self, other):
        return self.graph.add(other, self)













#







    def __rsub__(self, other):
        return self.graph.sub(other, self)













#







    def __rmul__(self, other):
        return self.graph.mul(other, self)













#







    def __rtruediv__(self, other):
        return self.graph.div(other, self)