Extract coefficients from a based on t and reshape to make it broadcastable with x_shape.
(a, t, x_shape)
source not stored for this graph (policy: none)
no outgoing calls
no test coverage detected