# This file implements acceleration/velocity calculation

import torch
import torch.nn as nn


class AcFusionLayer(nn.Module):
	"""docstring for AcFusionLayer"""
	def __init__(self, ):
		super(AcFusionLayer, self).__init__()
	
	def forward(self, flo10, flo12, t=0.5):
		"""
			-- input: four flows
			-- output: center shift
		"""

		return 0.5 * ((t + t**2)*flo12 - (t - t**2)*flo10)
		

# This file implements acceleration/velocity calculation

import torch
import torch.nn as nn


class AcFusionLayer(nn.Module):
	"""docstring for AcFusionLayer"""
	def __init__(self, ):
		super(AcFusionLayer, self).__init__()
	
	def forward(self, flo10, flo12, t=0.5):
		"""
			-- input: four flows
			-- output: center shift
		"""

		return 0.5 * ((t + t**2)*flo12 - (t - t**2)*flo10)
		

# This file implements acceleration/velocity calculation docstring for AcFusionLayer -- input: four flows 			-- output: center shift