Как я могу изменить следующий код с pytorch на tenorflow?

Question

Я хочу изменить следующую сеть Pytorch (v1.2) на tenorflow. Я путаю между tf.nn.conv2d и tf.keras.layers.Conv2D что мне выбрать?

import torch.nn as nn  
nn.Sequential(nn.Conv2d(in_planes, out_planes, kernel_size=kernel_size, stride=stride, padding=padding, dilation=dilation, bias=True), 
    nn.BatchNorm2d(out_planes), 
    nn.ReLU(inplace=True))

Answer 1

tf.nn.conv2d - функциональный API, а tf.keras.layers.Conv2D - API класса слоя. Вы должны использовать последний. Это похоже на отношения между torch.nn.functional.conv2d и torch.nn.Conv2D.

import tensorflow as tf
from tensorflow.keras import Sequential
from tensorflow.keras.layers import Conv2D, ReLU, BatchNormalization

model = Sequential()
model.add(Conv2D(filters=10, kernel_size=3, strides=1))
model.add(BatchNormalization())
model.add(ReLU())