Question

是tensorflow的新手，想将pytorch代码转换成tensorflow代码。如何执行呢？

X_train, Y_train = Variable(torch.Tensor(X[:-30])), Variable(torch.Tensor(Y[:-30]))
X_test, Y_test = Variable(torch.Tensor(X[-30:])), Variable(torch.Tensor(Y[-30:]))

X_train = X_train.view(X_train.size(0), -1)
Y_train = Y_train.view(Y_train.size(0), -1)

data = torch.cat([X_train, Y_train], dim=1)
print(type(data))

Answer 1

您尝试过tf.concat（）吗？您可能会在https://www.tensorflow.org/api_docs/python/tf/concat上找到它，但我相信它适用于在一个维度上具有不同大小的张量，例如（10，10，3）和（10，10，8）可以串联为（10， 10、11）您也可以尝试tf.stack（）

张量流中不等维张量的连接

1 个答案: