google-deepmind / deepmind-research

This repository contains implementations and illustrative code to accompany DeepMind publications

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

tf.unique functionality changed

pjuangph opened this issue · comments

tf.unique finds unique entries in an array [1,2,3,4,5,5] It doesn't work for multi-dimensional arrays like what you are passing in here.

https://github.com/deepmind/deepmind-research/blob/11c2ab53e8afd24afa8904f22fd81b699bfbce6e/meshgraphnets/common.py#L46