17 lines
No EOL
374 B
Python
17 lines
No EOL
374 B
Python
def prepare_data(batch):
|
|
|
|
phi_ini = batch['input'][:, [1]]
|
|
D1 = batch['style'][:, [0]]
|
|
D2 = batch['style'][:, [1]]
|
|
gravpot = batch['target'][:, [0]]
|
|
|
|
_input = batch['input']
|
|
_target = (gravpot/D1 - phi_ini)/D1
|
|
_style = batch['style']
|
|
|
|
return {
|
|
'input': _input,
|
|
'target': _target,
|
|
'style': _style
|
|
}
|
|
|