:param target: [ ..., J, d] dtype=float :param logits: [ ..., J], dtype=float :param mask: [ ..., J], dtype=bool :param scope: :return: [..., d], dtype=float
(target, logits, mask=None, scope=None)
source not stored for this graph (policy: none)
no test coverage detected