具有张量流的火炬的torch.cat等价是什么？

Question

def cxcy_to_xy(cxcy):
    """
    Convert bounding boxes from center-size coordinates (c_x, c_y, w, h) to boundary coordinates (x_min, y_min, x_max, y_max).

    :param cxcy: bounding boxes in center-size coordinates, a tensor of size (n_boxes, 4)
    :return: bounding boxes in boundary coordinates, a tensor of size (n_boxes, 4)
    """
    return torch.cat([cxcy[:, :2] - (cxcy[:, 2:] / 2),  # x_min, y_min
                      cxcy[:, :2] + (cxcy[:, 2:] / 2)], 1)  # x_max, y_max

我要更改这个带有tensorflow 2.0的torch.cat

Answer 1

根据您所使用的TF中的API，很少有选项：

• tf.concat-最类似于tf.concat：

  torch.cat

• tf.concat(values, axis, name='concat') -如果您使用的是Keras顺序API：

  tf.keras.layers.concatenate

• tf.keras.layers.concatenate-如果您使用的是Keras功能性API：

  tf.keras.layers.concatenate(values, axis=-1, **kwargs)
  

如果使用Keras API，tf.keras.layers.Concatenate对于了解所有Keras串联函数之间的差异是很有帮助的。