Struct autograd::tensor::TensorCore

pub struct TensorCore {
    pub op: Box<Op>,
    pub inputs: Vec<Tensor>,
    pub top_rank: usize,
    pub shape: Option<Tensor>,
    pub persistent_array: Option<PersistentArray>,
    pub resource_lookup_key: Cell<usize>,
    pub is_placeholder: bool,
    pub has_gradient: bool,
    pub input_indices: Vec<usize>,
}

Fields

op: Box<Op>

Operation of this node.

inputs: Vec<Tensor>

References to immediate predecessors.

top_rank: usize

Rank number for topological ordering in a graph.

shape: Option<Tensor>

Symbolic shape of this tensor.

persistent_array: Option<PersistentArray>

Variable or constant array is placed here.

resource_lookup_key: Cell<usize>

Used to look up a resource of this tensor.

is_placeholder: bool

Immutable flag of tensor is placeholder or not.

has_gradient: bool

op can have gradient?

input_indices: Vec<usize>

Indices of arrays used in compute

Trait Implementations