我如何将下面的代码从pytorch更改为tensorflow？
提问于 2019-10-23 17:38:24
我想将下面的py手电筒网络(v1.2)更改为tensorflow。我在tf.keras.layers.Conv2D和tf.nn.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))
回答 1
Stack Overflow用户
发布于 2019-10-23 18:56:12
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())页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/58532977
复制相关文章
腾讯云开发者
