When axis=1, tf.gather gathered 3rd and 1st column.,This is another example equivalent to torch.gather when dim=1,Here is the example of ...
確定! 回上一頁