MCPcopy
hub / github.com/dmlc/dgl / _expand

Function _expand

python/dgl/backend/tensorflow/sparse.py:135–136  ·  view source on GitHub ↗
(x, shape)

Source from the content-addressed store, hash-verified

133
134
135def _expand(x, shape):
136 return tf.broadcast_to(x, (x.shape[0], *shape))
137
138
139def gspmm_real(gidx, op, reduce_op, X, Y):

Callers 1

gradFunction · 0.70

Calls

no outgoing calls

Tested by

no test coverage detected