MAML核心代码

 import numpy as np

    
 import tensorflow as tf
    
  
    
 class MAML:
    
 	def __init__(self, d, c, nway, meta_lr=1e-3, train_lr=1e-2):
    
 		"""
    
 		定义了图片大小、通道、样本类way、学习率
    
 		:param d:图片大小
    
 		:param c:通道
    
 		:param nway:类数
    
 		:param meta_lr:
    
 		:param train_lr:
    
 		"""
    
 		self.d = d
    
 		self.c = c
    
 		self.nway = nway
    
 		self.meta_lr = meta_lr
    
 		self.train_lr = train_lr
    
  
    
 		print('img shape:', self.d, self.d, self.c, 'meta-lr:', meta_lr, 'train-lr:', train_lr)
    
  
    
 	def build(self, support_xb, support_y