Compute block diagonal matrix multiplication efficiently using broadcasting Last dimension will be used for matrix multiplication. prev_layer.output_shape = N x D_1 x D_2 x ... x D_{n-1} x D_n output_shape = N x D_1 x D_2 x ... x D_{n-1} x n_out
source not stored for this graph (policy: none)
nothing calls this directly
no outgoing calls
no test coverage detected