Сериализировать Тензор - PullRequest
       1

Сериализировать Тензор

0 голосов
/ 14 февраля 2019

У меня есть такой тензор, как этот

Tensor("activation_1/Relu:0", shape=(?, 32, 32, 96), dtype=float32)

Я хочу сериализовать этот тензор, чтобы переместить его через устройства, а затем восстановить его, но, кажется, невозможно сериализовать такой, используяPickle or Dill

Update

Тензор является результатом этого блока модели.

`def conv_module(x, K, kX, kY, stride, padding="same"):
    # define a CONV => BN => RELU pattern
    x = Conv2D(K, (kX, kY), strides=stride, padding=padding)(x)
    x = BatchNormalization()(x)
    x = Activation("relu")(x)
    # return the block  
    return x`

В настоящее время я не использую нетерпеливое выполнение, но я мог бы сделать это также, если бы это было возможно.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...