mindearth.cell.DgmrDiscriminator
- class mindearth.cell.DgmrDiscriminator(in_channels=1, num_spatial_frames=8, conv_type='standard')[source]
The Dgmr Discriminator is based on Temporal Discriminator and Spatial Discriminator, which contains deep residual block. The details can be found in Skilful precipitation nowcasting using deep generative models of radar.
- Parameters
- Inputs:
x (Tensor) - Tensor of shape
.
- Outputs:
Tensor, the output of the DgmrDiscriminator.
output (Tensor) - Tensor of shape
.
- Supported Platforms:
Ascend
GPU
Examples
>>> import numpy as np >>> import mindspore as ms >>> from mindspore import ops, Tensor >>> from mindspore.nn import Cell >>> from mindearth.cell.dgmr.dgmrnet import DgmrDiscriminator >>> real_and_generator = np.random.rand(2, 22, 1, 256, 256).astype(np.float32) >>> net = DgmrDiscriminator(in_channels=1, num_spatial_frames=8, con_type="standard") >>> out = net(Tensor(real_and_generator, ms.float32)) >>> print(out.shape) (2, 2, 1)