tf.nn.conv2d初探

一、前因

打算自己写代码，实现cifar10数据的cnn，在码代码的过程中，发现对tf.nn.conv2d等函数的处理结果不甚了解，也不清楚多通道时数据的计算方法，故做了以下测试

二、代码

环境：python3.6、tensorflow1.14

import pickle
import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt

#读取Cifar10的函数
def loadCifar10(path):
    with open(path,'rb') as fo:
        dict=pickle.load(fo,encoding='bytes')
    return dict


dict=loadCifar10('.\cifar-10-batches-py\data_batch_1')

data=dict[b'data']
labels=np.reshape(np.array(dict[b'labels']),[10000,1])

data0=data[0]  #读第0张图

input_data=tf